mnist can be load correctly and can be used to train