Compare commits

..
2 Commits
Author SHA1 Message Date
Phil Wang cf4f44909f 0.1.5 2022-12-21 08:35:59 -08:00
Phil Wang a7f2d670bb always set results_folder on Trainer 2022-12-21 08:34:34 -08:00
2 changed files with 3 additions and 3 deletions
@@ -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
+1 -1
View File
@@ -1 +1 @@
__version__ = '0.1.0'
__version__ = '0.1.5'