Training update

1 files changed, 1 insertions, 3 deletions

diff --git a/training/functional.py b/training/functional.py
index 1548784..3d27380 100644
--- a/training/functional.py
+++ b/training/functional.py
@@ -87,8 +87,6 @@ def save_samples(
 ):
     print(f"Saving samples for step {step}...")
 
-    samples_path = output_dir.joinpath("samples")
-
     grid_cols = min(batch_size, 4)
     grid_rows = (num_batches * batch_size) // grid_cols
 
@@ -120,7 +118,7 @@ def save_samples(
 
     for pool, data, gen in datasets:
         all_samples = []
-        file_path = samples_path.joinpath(pool, f"step_{step}.jpg")
+        file_path = output_dir.joinpath(pool, f"step_{step}.jpg")
         file_path.parent.mkdir(parents=True, exist_ok=True)
 
         batches = list(itertools.islice(itertools.cycle(data), batch_size * num_batches))