diff options
Diffstat (limited to 'train_ti.py')
-rw-r--r-- | train_ti.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/train_ti.py b/train_ti.py index 9bc74c1..ef71f6f 100644 --- a/train_ti.py +++ b/train_ti.py | |||
@@ -188,6 +188,12 @@ 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", | ||
192 | type=float, | ||
193 | default=0.1, | ||
194 | help="Perlin offset noise strength.", | ||
195 | ) | ||
196 | parser.add_argument( | ||
191 | "--num_train_epochs", | 197 | "--num_train_epochs", |
192 | type=int, | 198 | type=int, |
193 | default=100 | 199 | default=100 |
@@ -655,6 +661,7 @@ def main(): | |||
655 | checkpoint_frequency=args.checkpoint_frequency, | 661 | checkpoint_frequency=args.checkpoint_frequency, |
656 | milestone_checkpoints=not args.no_milestone_checkpoints, | 662 | milestone_checkpoints=not args.no_milestone_checkpoints, |
657 | global_step_offset=global_step_offset, | 663 | global_step_offset=global_step_offset, |
664 | perlin_strength=args.perlin_strength, | ||
658 | # -- | 665 | # -- |
659 | tokenizer=tokenizer, | 666 | tokenizer=tokenizer, |
660 | sample_scheduler=sample_scheduler, | 667 | sample_scheduler=sample_scheduler, |