wandb integration
AK391 opened this issue · comments
Ahsen Khaliq commented
for the finetuning stage in colab, wandb is useful to track metrics, see https://github.com/danielroich/PTI#weights-and-biases for example. This would be helpful when running multiple experiments in colab, if you have time, otherwise I can also look into this as a PR. Thanks
Min Jin Chong commented
I will take a look at it in a day or two! What metrics will be useful for you? Loss and current generation?
Ahsen Khaliq commented
I will take a look at it in a day or two! What metrics will be useful for you? Loss and current generation?
yes loss and current generation would be great, thank you
Min Jin Chong commented
Done!