How to log the number of model parameters to W&B through the Trainer?

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?