Skip to content

Commit 4b8a192

Browse files
committed
add optimizer save option to shared.opts
1 parent 17a2076 commit 4b8a192

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

modules/shared.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -286,6 +286,7 @@ def options_section(section_identifier, options_dict):
286286

287287
options_templates.update(options_section(('training', "Training"), {
288288
"unload_models_when_training": OptionInfo(False, "Move VAE and CLIP to RAM when training hypernetwork. Saves VRAM."),
289+
"save_optimizer_state": OptionInfo(False, "Saves Optimizer state with checkpoints. This will cause file size to increase VERY much."),
289290
"dataset_filename_word_regex": OptionInfo("", "Filename word regex"),
290291
"dataset_filename_join_string": OptionInfo(" ", "Filename join string"),
291292
"training_image_repeats_per_epoch": OptionInfo(1, "Number of repeats for a single input image per epoch; used only for displaying epoch number", gr.Number, {"precision": 0}),

0 commit comments

Comments
 (0)