From a7f2d670bbaad3dd4bc4d6375b6966b3f61f49a3 Mon Sep 17 00:00:00 2001 From: Phil Wang Date: Wed, 21 Dec 2022 08:34:34 -0800 Subject: [PATCH] always set results_folder on Trainer --- denoising_diffusion_pytorch/denoising_diffusion_pytorch.py | 4 ++-- denoising_diffusion_pytorch/version.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py b/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py index f67e404..708e09e 100644 --- a/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py +++ b/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py @@ -810,8 +810,8 @@ class Trainer(object): if self.accelerator.is_main_process: self.ema = EMA(diffusion_model, beta = ema_decay, update_every = ema_update_every) - self.results_folder = Path(results_folder) - self.results_folder.mkdir(exist_ok = True) + self.results_folder = Path(results_folder) + self.results_folder.mkdir(exist_ok = True) # step counter state diff --git a/denoising_diffusion_pytorch/version.py b/denoising_diffusion_pytorch/version.py index b794fd4..df9144c 100644 --- a/denoising_diffusion_pytorch/version.py +++ b/denoising_diffusion_pytorch/version.py @@ -1 +1 @@ -__version__ = '0.1.0' +__version__ = '0.1.1'