From dde18cd75e8902efbf3d99059ba67687b761f031 Mon Sep 17 00:00:00 2001 From: Shen-Chenhui Date: Fri, 19 Apr 2024 14:34:18 +0800 Subject: [PATCH] debug --- configs/vae_magvit_v2/inference/16x128x128.py | 1 + configs/vae_magvit_v2/inference/pipeline_16x128x128.py | 1 + 2 files changed, 2 insertions(+) diff --git a/configs/vae_magvit_v2/inference/16x128x128.py b/configs/vae_magvit_v2/inference/16x128x128.py index 5d25983..28b8f01 100644 --- a/configs/vae_magvit_v2/inference/16x128x128.py +++ b/configs/vae_magvit_v2/inference/16x128x128.py @@ -7,6 +7,7 @@ is_vae = True # Define dataset root = None data_path = "CSV_PATH" +max_test_samples = -1 use_image_transform = False num_workers = 4 video_contains_first_frame = False diff --git a/configs/vae_magvit_v2/inference/pipeline_16x128x128.py b/configs/vae_magvit_v2/inference/pipeline_16x128x128.py index a67ea55..24ad8f6 100644 --- a/configs/vae_magvit_v2/inference/pipeline_16x128x128.py +++ b/configs/vae_magvit_v2/inference/pipeline_16x128x128.py @@ -8,6 +8,7 @@ use_pipeline = True # Define dataset root = None data_path = "CSV_PATH" +max_test_samples = -1 use_image_transform = False num_workers = 4 video_contains_first_frame = False