fix: add missing import for time module in training loop
Browse files- train_abuse_model.py +3 -3
train_abuse_model.py
CHANGED
@@ -3,7 +3,7 @@
|
|
3 |
|
4 |
import logging
|
5 |
import io
|
6 |
-
import
|
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=
|
296 |
-
per_device_eval_batch_size=
|
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",
|