diff options
Diffstat (limited to 'train_ti.py')
-rw-r--r-- | train_ti.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/train_ti.py b/train_ti.py index ef71f6f..bbc5524 100644 --- a/train_ti.py +++ b/train_ti.py | |||
@@ -188,9 +188,9 @@ def parse_args(): | |||
188 | help='Vector shuffling algorithm. Choose between ["all", "trailing", "leading", "between", "auto", "off"]', | 188 | help='Vector shuffling algorithm. Choose between ["all", "trailing", "leading", "between", "auto", "off"]', |
189 | ) | 189 | ) |
190 | parser.add_argument( | 190 | parser.add_argument( |
191 | "--perlin_strength", | 191 | "--offset_noise_strength", |
192 | type=float, | 192 | type=float, |
193 | default=0.1, | 193 | default=0.15, |
194 | help="Perlin offset noise strength.", | 194 | help="Perlin offset noise strength.", |
195 | ) | 195 | ) |
196 | parser.add_argument( | 196 | parser.add_argument( |
@@ -661,7 +661,7 @@ def main(): | |||
661 | checkpoint_frequency=args.checkpoint_frequency, | 661 | checkpoint_frequency=args.checkpoint_frequency, |
662 | milestone_checkpoints=not args.no_milestone_checkpoints, | 662 | milestone_checkpoints=not args.no_milestone_checkpoints, |
663 | global_step_offset=global_step_offset, | 663 | global_step_offset=global_step_offset, |
664 | perlin_strength=args.perlin_strength, | 664 | offset_noise_strength=args.offset_noise_strength, |
665 | # -- | 665 | # -- |
666 | tokenizer=tokenizer, | 666 | tokenizer=tokenizer, |
667 | sample_scheduler=sample_scheduler, | 667 | sample_scheduler=sample_scheduler, |