Update lib/code_reviewer.py
Browse files- lib/code_reviewer.py +1 -0
lib/code_reviewer.py
CHANGED
@@ -83,6 +83,7 @@ class CodeReviewer:
|
|
83 |
output_dir (str): Directory where the fine-tuned model will be saved.
|
84 |
"""
|
85 |
training_args = TrainingArguments(
|
|
|
86 |
output_dir=output_dir,
|
87 |
per_device_train_batch_size=4,
|
88 |
num_train_epochs=3,
|
|
|
83 |
output_dir (str): Directory where the fine-tuned model will be saved.
|
84 |
"""
|
85 |
training_args = TrainingArguments(
|
86 |
+
report_to=[],
|
87 |
output_dir=output_dir,
|
88 |
per_device_train_batch_size=4,
|
89 |
num_train_epochs=3,
|