summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--train_ti.py2
1 files changed, 1 insertions, 1 deletions
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():
974 974
975 try: 975 try:
976 for epoch in range(num_epochs): 976 for epoch in range(num_epochs):
977 if accelerator.is_main_process and False: 977 if accelerator.is_main_process:
978 if epoch % args.sample_frequency == 0: 978 if epoch % args.sample_frequency == 0:
979 checkpointer.save_samples(global_step + global_step_offset, args.sample_steps) 979 checkpointer.save_samples(global_step + global_step_offset, args.sample_steps)
980 980