Minor Consistency Comments

[QiyaoWei]

Hey guys,

Thanks a lot for the brilliant repo! Just two minor comments regarding consistency---

1. There is a typo in the front page "Preliminaries" part, "export DATASET" should be "export DATASETS"
2. I feel like this is nitpicking, but bear with my OCD here. The printed loss during training is not exactly in the order of "train_loss, test_loss, train_acc, test_acc" :)

Thanks again for the excellent repo and paper, and congratulations on the ICLR acceptance!