I am using hugging face transforer to fine tune a Bert model. How to show f1 with wandb?
training_args = TrainingArguments(
output_dir='./results_'+folder+'/', # output directory
num_train_epochs = 3, # total # of training epochs
per_device_train_batch_size = n_batch, # batch size per device during training
per_device_eval_batch_size = n_batch, # batch size for evaluation
weight_decay = 0.01, # strength of weight decay
logging_dir ='./logs', # directory for storing logs
learning_rate = lr,
#warmup_steps = 1000, # number of warmup steps for learning rate scheduler
load_best_model_at_end = True,
evaluation_strategy = 'steps',
metric_for_best_model='accuracy',
report_to="wandb",
)
does not work