learning_by_grad_by_grad_repro icon indicating copy to clipboard operation
learning_by_grad_by_grad_repro copied to clipboard

Published 20 hours ago • verified publisher icon AdrienLE

Application to new dataset

Open thomasarmstrong98 opened this issue 5 years ago • 0 comments

Hi,

I tried applying the MNIST code to FashionMNIST however the code does not seem to train the LSMT optimizer properly.

To change to the FashionMNIST dataset, all I changed in the code was the line:

"dataset = datasets.FashionMNIST( './Fashion-MNIST', train=True, download=True, transform=torchvision.transforms.ToTensor() )"

Do you happen to know how this could be fixed?

Thanks, Tom

thomasarmstrong98 avatar Dec 10 '19 16:12 thomasarmstrong98