错误:
File “E2Capsnet.py”, line 397, in test
target = torch.sparse.torch.eye(NUM_CLASSES).index_select(dim=0, index=label)
AttributeError: module ‘torch.sparse’ has no attribute ‘torch’
原版程序:
target = torch.sparse.torch.eye(NUM_CLASSES).index_select(dim=0, index=label)
改动后的程序:
target = torch.eye(NUM_CLASSES).index_select(dim=0, index=label)
OK!