From 71b676060a484fe462c13ab2d487299ada5e4a5a Mon Sep 17 00:00:00 2001 From: VitekDev Date: Thu, 15 Feb 2024 13:01:07 +0000 Subject: [PATCH] allow using 0.01 in save frequncy for large datasets --- src/webui.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/webui.py b/src/webui.py index 560453d..de6ab3c 100755 --- a/src/webui.py +++ b/src/webui.py @@ -553,7 +553,7 @@ def setup_gradio(): TRAINING_SETTINGS["batch_size"] = gr.Number(label="Batch Size", value=128, precision=0) TRAINING_SETTINGS["gradient_accumulation_size"] = gr.Number(label="Gradient Accumulation Size", value=4, precision=0) with gr.Row(): - TRAINING_SETTINGS["save_rate"] = gr.Number(label="Save Frequency (in epochs)", value=5, precision=0) + TRAINING_SETTINGS["save_rate"] = gr.Number(label="Save Frequency (in epochs)", value=5, precision=2) TRAINING_SETTINGS["validation_rate"] = gr.Number(label="Validation Frequency (in epochs)", value=5, precision=0) with gr.Row():