From 1fdf3688a30a9370761b148c42c5653124f735d9 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Mon, 2 Jan 2023 17:10:28 +0100 Subject: Save args before training, too --- train_ti.py | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'train_ti.py') diff --git a/train_ti.py b/train_ti.py index 5cbef06..6c74854 100644 --- a/train_ti.py +++ b/train_ti.py @@ -903,6 +903,10 @@ def main(): ) global_progress_bar.set_description("Total progress") + save_args(basepath, args, { + "global_step": global_step + global_step_offset + }) + try: for epoch in range(num_epochs): if accelerator.is_main_process: -- cgit v1.2.3-54-g00ecf