andrembcosta's picture
Training complete on TPU - full dataset - lr=1e-4
02ef0c2 verified