using train_one_step(), you will get the behavior of automatic disk syncs at the end of training.
Attach a file by drag & drop or click to upload
