diff options
| author | Josh R <josh@joshrubenstein.com> | 2023-01-13 01:31:05 +0000 |
|---|---|---|
| committer | Josh R <josh@joshrubenstein.com> | 2023-01-13 01:31:05 +0000 |
| commit | 0b262802b86a55c4f71faf377f2cb1aee2960b63 (patch) | |
| tree | f3b9b206d822d9f32b99fdc55826ead6fbca3757 /modules/textual_inversion/logging.py | |
| parent | d7aec59c4eb02f723b3d55c6f927a42e97acd679 (diff) | |
| download | stable-diffusion-webui-gfx803-0b262802b86a55c4f71faf377f2cb1aee2960b63.tar.gz stable-diffusion-webui-gfx803-0b262802b86a55c4f71faf377f2cb1aee2960b63.tar.bz2 stable-diffusion-webui-gfx803-0b262802b86a55c4f71faf377f2cb1aee2960b63.zip | |
add gradient settings to training settings log files
Diffstat (limited to 'modules/textual_inversion/logging.py')
| -rw-r--r-- | modules/textual_inversion/logging.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/textual_inversion/logging.py b/modules/textual_inversion/logging.py index 8b1981d5..31e50b64 100644 --- a/modules/textual_inversion/logging.py +++ b/modules/textual_inversion/logging.py @@ -2,7 +2,7 @@ import datetime import json
import os
-saved_params_shared = {"model_name", "model_hash", "initial_step", "num_of_dataset_images", "learn_rate", "batch_size", "data_root", "log_directory", "training_width", "training_height", "steps", "create_image_every", "template_file"}
+saved_params_shared = {"model_name", "model_hash", "initial_step", "num_of_dataset_images", "learn_rate", "batch_size", "clip_grad_mode", "clip_grad_value", "gradient_step", "data_root", "log_directory", "training_width", "training_height", "steps", "create_image_every", "template_file"}
saved_params_ti = {"embedding_name", "num_vectors_per_token", "save_embedding_every", "save_image_with_stored_embedding"}
saved_params_hypernet = {"hypernetwork_name", "layer_structure", "activation_func", "weight_init", "add_layer_norm", "use_dropout", "save_hypernetwork_every"}
saved_params_all = saved_params_shared | saved_params_ti | saved_params_hypernet
|