I’m using HF’s Trainer
with the TrainingArguments
’s parameter report_to
argument set to wandb
. How can I specify that I want the total number of parameters in my model logged at the beginning? Or do I just need to do this manually myself?
I’m using HF’s Trainer
with the TrainingArguments
’s parameter report_to
argument set to wandb
. How can I specify that I want the total number of parameters in my model logged at the beginning? Or do I just need to do this manually myself?