global_step=3783, training_loss=0.2294789322652169, metrics={'train_loss': 0.2294789322652169})
https://colab.research.google.com/drive/1iZRjkTm7Sv3_JZJx2MAk384hQuk-fq1W?usp=sharing