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_dreambooth.py | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'train_dreambooth.py') diff --git a/train_dreambooth.py b/train_dreambooth.py index 05f6cb5..cd0bf67 100644 --- a/train_dreambooth.py +++ b/train_dreambooth.py @@ -955,6 +955,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