From 9bfb4a078f63a7ce6e35e89093f17febd9ff4b51 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Fri, 24 Mar 2023 17:23:09 +0100 Subject: Update --- train_ti.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'train_ti.py') 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(): help='Vector shuffling algorithm. Choose between ["all", "trailing", "leading", "between", "auto", "off"]', ) parser.add_argument( - "--perlin_strength", + "--offset_noise_strength", type=float, - default=0.1, + default=0.15, help="Perlin offset noise strength.", ) parser.add_argument( @@ -661,7 +661,7 @@ def main(): checkpoint_frequency=args.checkpoint_frequency, milestone_checkpoints=not args.no_milestone_checkpoints, global_step_offset=global_step_offset, - perlin_strength=args.perlin_strength, + offset_noise_strength=args.offset_noise_strength, # -- tokenizer=tokenizer, sample_scheduler=sample_scheduler, -- cgit v1.2.3-54-g00ecf