From 2dfd1790078753f19ca8c585ac77079f3114f3a9 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Mon, 16 Jan 2023 21:47:06 +0100 Subject: Training update --- training/functional.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) (limited to 'training/functional.py') 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)) -- cgit v1.2.3-54-g00ecf