wandb integration

[AK391]

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

[mchong6]

I will take a look at it in a day or two! What metrics will be useful for you? Loss and current generation?

[AK391]

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

[mchong6]

Done!