diff options
author | Volpeon <git@volpeon.ink> | 2023-03-28 16:28:44 +0200 |
---|---|---|
committer | Volpeon <git@volpeon.ink> | 2023-03-28 16:28:44 +0200 |
commit | b59611506d2eac1b280ada07db0bf9de15b41a26 (patch) | |
tree | 3f4a1e49f0db72443e45bc64d2c5b9d99ce055fc /train_lora.py | |
parent | Support num_train_steps arg again (diff) | |
download | textual-inversion-diff-b59611506d2eac1b280ada07db0bf9de15b41a26.tar.gz textual-inversion-diff-b59611506d2eac1b280ada07db0bf9de15b41a26.tar.bz2 textual-inversion-diff-b59611506d2eac1b280ada07db0bf9de15b41a26.zip |
Fix
Diffstat (limited to 'train_lora.py')
-rw-r--r-- | train_lora.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/train_lora.py b/train_lora.py index a9c6e52..6f8644b 100644 --- a/train_lora.py +++ b/train_lora.py | |||
@@ -630,7 +630,7 @@ def main(): | |||
630 | num_train_epochs = args.num_train_epochs | 630 | num_train_epochs = args.num_train_epochs |
631 | 631 | ||
632 | if num_train_epochs is None: | 632 | if num_train_epochs is None: |
633 | num_train_epochs = math.ceil(len(datamodule.train_dataset) / args.num_train_steps) | 633 | num_train_epochs = math.ceil(args.num_train_steps / len(datamodule.train_dataset)) |
634 | 634 | ||
635 | optimizer = create_optimizer( | 635 | optimizer = create_optimizer( |
636 | itertools.chain( | 636 | itertools.chain( |