CapsNet-pytorch
CapsNet-pytorch copied to clipboard
Bug fix and change of defaults
- fix a mismatch between default and actual value of
lrparam. - some changes to default values for low memory.
- compatibility change from
.data[0]toitem()for newer PyTorch versions.