From b87e46095bcee7b5ee52a1ff06691833fe31df8c Mon Sep 17 00:00:00 2001 From: unknown Date: Sat, 19 Oct 2024 00:14:13 +0300 Subject: [PATCH] fix max_sample --- finetune_gradio.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/finetune_gradio.py b/finetune_gradio.py index d6db8cc..3f442dc 100644 --- a/finetune_gradio.py +++ b/finetune_gradio.py @@ -294,7 +294,9 @@ def start_training(dataset_name="", f"--last_per_steps {last_per_steps} " \ f"--dataset_name {dataset_name}" if finetune:cmd += f" --finetune {finetune}" - print(cmd) + + print(cmd) + try: # Start the training process training_process = subprocess.Popen(cmd, shell=True) @@ -525,6 +527,8 @@ def calculate_train(name_project,batch_size_type,max_samples,learning_rate,num_w if samples<64: max_samples = int(samples * 0.25) + else: + max_samples = 64 num_warmup_updates = int(samples * 0.10) save_per_updates = int(samples * 0.25)