CapsNet-pytorch icon indicating copy to clipboard operation
CapsNet-pytorch copied to clipboard

Bug fix and change of defaults

Open comsaint opened this issue 4 years ago • 0 comments

  1. fix a mismatch between default and actual value of lr param.
  2. some changes to default values for low memory.
  3. compatibility change from .data[0] to item() for newer PyTorch versions.

comsaint avatar Apr 13 '21 11:04 comsaint