Wandb not logging metrics

[gamayos]

Hello,

In provided Google colab script, enabling Wandb logging is working fine for system and text logs, but not the metrics. I've seen this was an issue in the past but seems to have been resolved.

Thank you in advance and keep up your great work!