diff options
| author | Volpeon <git@volpeon.ink> | 2023-01-02 11:27:38 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-01-02 11:27:38 +0100 |
| commit | c24a24c77a825e71de2e67c1515c84d2b77701fa (patch) | |
| tree | 2ed4616191c8f61c3f502375bd6698e0d29d041e /train_ti.py | |
| parent | Update (diff) | |
| download | textual-inversion-diff-c24a24c77a825e71de2e67c1515c84d2b77701fa.tar.gz textual-inversion-diff-c24a24c77a825e71de2e67c1515c84d2b77701fa.tar.bz2 textual-inversion-diff-c24a24c77a825e71de2e67c1515c84d2b77701fa.zip | |
Improved one cycle scheduler
Diffstat (limited to 'train_ti.py')
| -rw-r--r-- | train_ti.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/train_ti.py b/train_ti.py index 870bd40..775b918 100644 --- a/train_ti.py +++ b/train_ti.py | |||
| @@ -250,7 +250,7 @@ def parse_args(): | |||
| 250 | parser.add_argument( | 250 | parser.add_argument( |
| 251 | "--lr_annealing_exp", | 251 | "--lr_annealing_exp", |
| 252 | type=int, | 252 | type=int, |
| 253 | default=2, | 253 | default=1, |
| 254 | help='If lr_annealing_func is "half_cos" or "cos", exponent to modify the function' | 254 | help='If lr_annealing_func is "half_cos" or "cos", exponent to modify the function' |
| 255 | ) | 255 | ) |
| 256 | parser.add_argument( | 256 | parser.add_argument( |