rdevon / cortex

A machine learning library for PyTorch
BSD 3-Clause "New" or "Revised" License
92 stars 25 forks source link

fixing mnist with new torchvision version #191

Closed rdevon closed 6 years ago

rdevon commented 6 years ago

Torchvision behavior for MNIST changed, and this PR brings cortex up-to-date with these changes

dmitriy-serdyuk commented 6 years ago

MNIST works now, but VAE is broken.

RuntimeError: input and target shapes do not match: input [640 x 1 x 24 x 24], target [640 x 1 x 28 x 28]
rdevon commented 6 years ago

@dmitriy-serdyuk So this is because the default convnet doesn't work for MNIST. You need to use the arguments --encoder_type mnist --decoder_type mnist