Update gradio demo

This commit is contained in:
Edresson Casanova 2023-11-24 15:40:35 -03:00
parent 8967fc7ef2
commit c76fb856d1
2 changed files with 2 additions and 2 deletions

View File

@ -23,7 +23,7 @@ def train_gpt(language, num_epochs, batch_size, train_csv, eval_csv, output_path
OPTIMIZER_WD_ONLY_ON_WEIGHTS = True # for multi-gpu training please make it False
START_WITH_EVAL = True # if True it will star with evaluation
BATCH_SIZE = batch_size # set here the batch size
GRAD_ACUMM_STEPS = 1 # set here the grad accumulation steps
GRAD_ACUMM_STEPS = 4 # set here the grad accumulation steps
# Note: we recommend that BATCH_SIZE * GRAD_ACUMM_STEPS need to be at least 252 for more efficient training. You can increase/decrease BATCH_SIZE but then set GRAD_ACUMM_STEPS accordingly.

View File

@ -189,7 +189,7 @@ with gr.Blocks() as demo:
minimum=2,
maximum=512,
step=1,
value=16,
value=4,
)
progress_train = gr.Label(
label="Progress:"