diff --git a/hrdae/__main__.py b/hrdae/__main__.py
index cd28e55..1dafb33 100644
--- 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
index 03bd36e..176503d 160000
--- a/hrdae/conf
+++ b/hrdae/conf
@@ -1 +1 @@
-Subproject commit 03bd36e839b6e37df34506588ab21d12bc6ed69e
+Subproject commit 176503d22324aecb00cffd7c3b24cd0b3f3beb6b
diff --git a/hrdae/models/functions.py b/hrdae/models/functions.py
index 8e782df..b007b81 100644
--- 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)