softened-similarity-learning
softened-similarity-learning copied to clipboard
test issue
It happened this issue when I was running test.py. Could you help me? File "test.py", line 80, in evaluate accuracy = (all_labels == test_labels).sum().item() / len(all_labels) RuntimeError: The size of tensor a (3368) must match the size of tensor b (19732) at non-singleton dimension 0
modify test.py line 75 change dim=1 to dim=0
It happened this issue when I was running test.py. Could you help me? File "test.py", line 80, in evaluate accuracy = (all_labels == test_labels).sum().item() / len(all_labels) RuntimeError: The size of tensor a (3368) must match the size of tensor b (19732) at non-singleton dimension 0
modify test.py line 75 change dim=1 to dim=0