yadnyeshkolte's picture
Train with torchvision MNIST and upgrade inference model
56bdf13