From d12634cb211e10223ad77a43a562d80678b2d662 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Wed, 11 Jan 2023 21:57:43 +0100 Subject: Heck --- train_ti.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'train_ti.py') diff --git a/train_ti.py b/train_ti.py index 1054a5d..e6c437e 100644 --- a/train_ti.py +++ b/train_ti.py @@ -974,7 +974,7 @@ def main(): try: for epoch in range(num_epochs): - if accelerator.is_main_process and False: + if accelerator.is_main_process: if epoch % args.sample_frequency == 0: checkpointer.save_samples(global_step + global_step_offset, args.sample_steps) -- cgit v1.2.3-54-g00ecf