From 9d279587b82260886480ef45844c25c50bf3371d Mon Sep 17 00:00:00 2001 From: Volpeon Date: Mon, 2 Jan 2023 17:02:41 +0100 Subject: Fix --- train_ti.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) (limited to 'train_ti.py') diff --git a/train_ti.py b/train_ti.py index f1dbed1..5cbef06 100644 --- a/train_ti.py +++ b/train_ti.py @@ -1006,7 +1006,7 @@ def main(): if accelerator.is_main_process: print("Finished! Saving final checkpoint and resume state.") checkpointer.checkpoint(global_step + global_step_offset, "end") - checkpointer.save_samples(global_step_offset, args.sample_steps) + checkpointer.save_samples(global_step + global_step_offset, args.sample_steps) save_args(basepath, args, { "global_step": global_step + global_step_offset }) @@ -1016,7 +1016,6 @@ def main(): if accelerator.is_main_process: print("Interrupted, saving checkpoint and resume state...") checkpointer.checkpoint(global_step + global_step_offset, "end") - checkpointer.save_samples(global_step_offset, args.sample_steps) save_args(basepath, args, { "global_step": global_step + global_step_offset }) -- cgit v1.2.3-54-g00ecf