wassname/denoising-diffusion-pytorch
1
0
Fork 0
mirror of https://github.com/wassname/denoising-diffusion-pytorch.git synced 2026-09-09 11:21:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

always set results_folder on Trainer

Browse Source
This commit is contained in:
Phil Wang
2022-12-21 08:34:34 -08:00
parent 6ab29d5cea
commit a7f2d670bb
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
denoising_diffusion_pytorch/denoising_diffusion_pytorch.py
+2 -2
View File
@@ -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
denoising_diffusion_pytorch/version.py
+1 -1
View File
@@ -1 +1 @@
__version__ = '0.1.0'
__version__ = '0.1.1'