finish modification

nnaakkaaii · Jun 16, 2024 · 47a1740 · 47a1740
1 parent 6e2e8e7
commit 47a1740
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/hrdae/models/functions.py b/hrdae/models/functions.py
@@ -56,8 +56,6 @@ def save_reconstructed_images(
     elif len(original.shape) == 6:
         b, _, _, d, h, w = original.shape
         for bi in range(0, b, 5):
-            print(original.shape)
-            print(reconstructed.shape)
             _save_images(
                 original[bi, :, :, :, :, w // 2],  # (t, 1, d, h)
                 reconstructed[bi, :, :, :, :, w // 2],  # (t, 1, d, h)

diff --git a/tuning/ct.py b/tuning/ct.py
@@ -265,7 +265,7 @@ def objective(trial):
         result_dir=result_dir,
         dataloader=dataloader_option,
         model=model_option,
-        n_epoch=1,
+        n_epoch=50,
     )
     result_dir.mkdir(parents=True, exist_ok=True)
     with open(result_dir / "config.json", "w") as f: