diff options
| author | AngelBottomless <35677394+aria1th@users.noreply.github.com> | 2022-10-29 16:36:43 +0900 |
|---|---|---|
| committer | aria1th <35677394+aria1th@users.noreply.github.com> | 2022-10-30 20:40:59 +0900 |
| commit | 4b8a192f680101de247dca79e48974b53bf961fe (patch) | |
| tree | b0d22c6e9fccf04f2d934c15761eaae7542260f7 /modules | |
| parent | 17a2076f72562b428052ee3fc8c43d19c03ecd1e (diff) | |
add optimizer save option to shared.opts
Diffstat (limited to 'modules')
| -rw-r--r-- | modules/shared.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/modules/shared.py b/modules/shared.py index e4f163c1..065b893d 100644 --- a/modules/shared.py +++ b/modules/shared.py @@ -286,6 +286,7 @@ options_templates.update(options_section(('system', "System"), { options_templates.update(options_section(('training', "Training"), {
"unload_models_when_training": OptionInfo(False, "Move VAE and CLIP to RAM when training hypernetwork. Saves VRAM."),
+ "save_optimizer_state": OptionInfo(False, "Saves Optimizer state with checkpoints. This will cause file size to increase VERY much."),
"dataset_filename_word_regex": OptionInfo("", "Filename word regex"),
"dataset_filename_join_string": OptionInfo(" ", "Filename join string"),
"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}),
|