search

adambielski / CapsNet-pytorch

PyTorch implementation of NIPS 2017 paper Dynamic Routing Between Capsules
BSD 3-Clause "New" or "Revised" License
490 stars 73 forks source link

Can't load model #2

Closed ryujaehun closed 6 years ago

ryujaehun commented 6 years ago

After the train, I encountered the following problems in the reconstruction_visualization process.

KeyError: 'unexpected key "conv1.weight" in state_dict'

ryujaehun commented 6 years ago

It was solved using --with_reconstruction