Merge branch 'main' of github.com:nnaakkaaii/D-3DVRVAE

nnaakkaaii · Jun 27, 2024 · bc6298a · bc6298a
2 parents 4b1c795 + 8ad3868
commit bc6298a
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/hrdae/__main__.py b/hrdae/__main__.py
@@ -15,8 +15,10 @@ def main(cfg: DictConfig):
         opt.experiment.result_dir
         / opt.experiment.dataloader.__class__.__name__
         / opt.experiment.model.__class__.__name__
-        / datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
     )
+    if hasattr(opt.experiment.model, "network"):
+        opt.experiment.result_dir /= opt.experiment.model.network.__class__.__name__
+    opt.experiment.result_dir /= datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
     save_options(opt, opt.experiment.result_dir)
     if (
         isinstance(opt.experiment, TrainExpOption)

diff --git a/hrdae/conf b/hrdae/conf
diff --git a/hrdae/models/functions.py b/hrdae/models/functions.py
@@ -69,7 +69,7 @@ def save_reconstructed_images(
         )
     elif len(original.shape) == 5:
         b, _, _, _, _ = original.shape
-        for bi in range(0, b, 5):
+        for bi in range(b):
             _save_images(
                 original[bi],  # (t, 1, h, w)
                 reconstructed[bi],  # (t, 1, h, w)
@@ -78,7 +78,7 @@ def save_reconstructed_images(
             )
     elif len(original.shape) == 6:
         b, _, _, d, h, w = original.shape
-        for bi in range(0, b, 5):
+        for bi in range(b):
             _save_images(
                 original[bi, :, :, :, :, w // 2],  # (t, 1, d, h)
                 reconstructed[bi, :, :, :, :, w // 2],  # (t, 1, d, h)