Saving model's weights

Hi,
I’m training a BERT model and I’m trying to save the weights to wandb’s files tab in the end of the training.
How can I accomplish that?
Also - how can I load the weights from wandb’s files tab?
I’m using this code:
https://wandb.ai/cayush/bert-finetuning/reports/Sentence-classification-with-Huggingface-BERT-and-W&B–Vmlldzo4MDMwNA

This topic was automatically closed 60 days after the last reply. New replies are no longer allowed.