diff options
| author | Melan <alexleander91@gmail.com> | 2022-10-20 22:43:08 +0200 |
|---|---|---|
| committer | Melan <alexleander91@gmail.com> | 2022-10-20 22:43:08 +0200 |
| commit | 7543cf5e3b5eaced00582da257801227d1ff2a6e (patch) | |
| tree | 76235f861bd022936931bfab4beb39555caee5ea | |
| parent | 8f5912984794c4c69e429c4636e984854d911b6a (diff) | |
Fixed some typos in the code
| -rw-r--r-- | modules/hypernetworks/hypernetwork.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/modules/hypernetworks/hypernetwork.py b/modules/hypernetworks/hypernetwork.py index 5e919775..0cd94f49 100644 --- a/modules/hypernetworks/hypernetwork.py +++ b/modules/hypernetworks/hypernetwork.py @@ -284,19 +284,19 @@ def train_hypernetwork(hypernetwork_name, learn_rate, batch_size, data_root, log last_saved_file = "<none>"
last_saved_image = "<none>"
- ititial_step = hypernetwork.step or 0
- if ititial_step > steps:
+ initial_step = hypernetwork.step or 0
+ if initial_step > steps:
return hypernetwork, filename
- scheduler = LearnRateScheduler(learn_rate, steps, ititial_step)
+ scheduler = LearnRateScheduler(learn_rate, steps, initial_step)
optimizer = torch.optim.AdamW(weights, lr=scheduler.learn_rate)
if shared.opts.training_enable_tensorboard:
tensorboard_writer = textual_inversion.tensorboard_setup(log_directory)
- pbar = tqdm.tqdm(enumerate(ds), total=steps - ititial_step)
+ pbar = tqdm.tqdm(enumerate(ds), total=steps - initial_step)
for i, entries in pbar:
- hypernetwork.step = i + ititial_step
+ hypernetwork.step = i + initial_step
scheduler.apply(optimizer, hypernetwork.step)
if scheduler.finished:
|