train_state.json

{"step": 26000, "val_acc": 0.9978392384694715, "val_loss": 0.01681531584014024}