Do you mind showing how to load the entire MNIST into gpu and train? pytorch tutorial doesn’t seem to have that.
I have successfully reduce the time to 1s with num_worker=6 but is it still 50% slower than mxnet.