mistralai / mistral-finetune

Apache License 2.0
2.45k stars 164 forks source link

Tensorboard integration #32

Open SaiKrishnaBala opened 1 month ago

SaiKrishnaBala commented 1 month ago

What configuration changes needs to be done in yaml so that it emits loss values which cab be fed to tensorboard. I have tried all the steps and i'm able to monitor the training loss values on wandb portal but i would like to have a local tensorbaord setup for this. Any references would be helpful.