I trained a model with these calls, but tensorboard files are not uploaded to the hub. I checked tensorboard is installed in the computer. Files are generated in the /runs folder. They’re just not included in the commit. Any idea why?
training_args = TrainingArguments(
output_dir=f"{model_name}-finetuned-imdb",
overwrite_output_dir=True,
evaluation_strategy="epoch",
learning_rate=2e-5,
weight_decay=0.01,
per_device_train_batch_size=batch_size,
per_device_eval_batch_size=batch_size,
push_to_hub=True,
fp16=True,
logging_steps=logging_steps,
report_to='tensorboard'
)
trainer = Trainer(
model=model,
args=training_args,
train_dataset=downsampled_dataset["train"],
eval_dataset=downsampled_dataset["test"],
data_collator=data_collator,
tokenizer=tokenizer,
)
trainer.train()
trainer.push_to_hub()