Skip to content

Commit a7f2d67

Browse files
committed
always set results_folder on Trainer
1 parent 6ab29d5 commit a7f2d67

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

denoising_diffusion_pytorch/denoising_diffusion_pytorch.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -810,8 +810,8 @@ def __init__(
810810
if self.accelerator.is_main_process:
811811
self.ema = EMA(diffusion_model, beta = ema_decay, update_every = ema_update_every)
812812

813-
self.results_folder = Path(results_folder)
814-
self.results_folder.mkdir(exist_ok = True)
813+
self.results_folder = Path(results_folder)
814+
self.results_folder.mkdir(exist_ok = True)
815815

816816
# step counter state
817817

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
__version__ = '0.1.0'
1+
__version__ = '0.1.1'

0 commit comments

Comments
 (0)