rshakked commited on
Commit
def4deb
·
1 Parent(s): c2281d2

fix: add missing import for time module in training loop

Browse files
Files changed (1) hide show
  1. train_abuse_model.py +3 -3
train_abuse_model.py CHANGED
@@ -3,7 +3,7 @@
3
 
4
  import logging
5
  import io
6
- import os
7
  import gradio as gr # ✅ required for progress bar
8
  from pathlib import Path
9
 
@@ -292,8 +292,8 @@ def run_training(progress=gr.Progress(track_tqdm=True)):
292
  training_args = TrainingArguments(
293
  output_dir="./results",
294
  num_train_epochs=3,
295
- per_device_train_batch_size=4,
296
- per_device_eval_batch_size=4,
297
  evaluation_strategy="epoch",
298
  save_strategy="epoch",
299
  logging_dir="./logs",
 
3
 
4
  import logging
5
  import io
6
+ import time
7
  import gradio as gr # ✅ required for progress bar
8
  from pathlib import Path
9
 
 
292
  training_args = TrainingArguments(
293
  output_dir="./results",
294
  num_train_epochs=3,
295
+ per_device_train_batch_size=8,
296
+ per_device_eval_batch_size=8,
297
  evaluation_strategy="epoch",
298
  save_strategy="epoch",
299
  logging_dir="./logs",