I have the same problem. I don’t see the loss reported. Using version 3.0.2. Here is my code:
training_args = TrainingArguments(
output_dir="./ClaimsBERTo_WordLevel",
overwrite_output_dir=False,
num_train_epochs=3,
per_gpu_train_batch_size=64,
save_steps=10,
logging_steps=10,
eval_steps=10,
save_total_limit=5,
evaluate_during_training=True,
do_eval=True,
logging_dir=’./logs’,
)
trainer = Trainer(
model=model,
args=training_args,
data_collator=data_collator,
train_dataset=train_dataset,
eval_dataset=val_dataset,
prediction_loss_only=True,
)
I hope someone will be able to help.
Thank you