On line 329 in lieConv.py the group object is not instantiated, causing problems with LieResNet's forward method. The obvious two-character fix seemed to get it working.
Ah yes, we always specified the argument in train_molec.py, train_img.py, train_springs.py so the default argument was never used. Thanks for pointing this out however and we'll change the default argument.
On line 329 in lieConv.py the group object is not instantiated, causing problems with LieResNet's
forward
method. The obvious two-character fix seemed to get it working.