From 94b676d91382267e7429bd68362019868affd9d1 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Mon, 13 Feb 2023 17:19:18 +0100 Subject: Update --- training/functional.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'training/functional.py') diff --git a/training/functional.py b/training/functional.py index ccbb4ad..83e70e2 100644 --- a/training/functional.py +++ b/training/functional.py @@ -129,7 +129,7 @@ def save_samples( for pool, data, gen in datasets: all_samples = [] - file_path = output_dir.joinpath(pool, f"step_{step}.jpg") + file_path = output_dir / 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