Actually, now I’m trying to converting tensorflow codes to pytorch.
I want to use all pytorch. but in Pytorch MNIST, train set and validation set is combined, and I don`t know how can split them exactly same as tensorflow.
As I know, in pytorch number of MNIST train, test set : [60000, 10000]
and in tensorflow, number of train, test, validation set : [55000, 10000, 5000]
so I tried with tensorflow MNIST dataset.