From 8d2aa65402c829583e26cdf2c336b8d3057657d6 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Fri, 5 May 2023 10:51:14 +0200 Subject: Update --- train_lora.py | 8 ++++++++ 1 file changed, 8 insertions(+) (limited to 'train_lora.py') diff --git a/train_lora.py b/train_lora.py index 3c8fc97..cc7c1ec 100644 --- a/train_lora.py +++ b/train_lora.py @@ -250,6 +250,12 @@ def parse_args(): default=0, help="Perlin offset noise strength.", ) + parser.add_argument( + "--input_pertubation", + type=float, + default=0, + help="The scale of input pretubation. Recommended 0.1." + ) parser.add_argument( "--num_train_epochs", type=int, @@ -1040,6 +1046,7 @@ def main(): checkpoint_output_dir=pti_checkpoint_output_dir, sample_frequency=pti_sample_frequency, offset_noise_strength=0, + input_pertubation=args.input_pertubation, no_val=True, ) @@ -1195,6 +1202,7 @@ def main(): checkpoint_output_dir=lora_checkpoint_output_dir, sample_frequency=lora_sample_frequency, offset_noise_strength=args.offset_noise_strength, + input_pertubation=args.input_pertubation, no_val=args.valid_set_size == 0, avg_loss=avg_loss, avg_acc=avg_acc, -- cgit v1.2.3-54-g00ecf