weird error occurs when run predict.py

[g1644222]

Sorry, execution of evaluate_ws.py failed with an error.
Can you guess the cause?

Thank you in advance.

algorithm	datasets	Accuracy
gloom	chaos	5
gloom	violin	6
gloom	tooth	5
brave	chaos	4
brave	violin	8
brave	tooth	9

question:
Is the accuracy of the algorithm gloom in the dataset chaos smaller than the accuracy of the algorithm brave in the dataset tooth?
error:
invalid argument 0: Sizes of tensors must match except in dimension 2. Got 3 and 4 in dimension 1 at /Users/soumith/code/builder/wheel/pytorch-src/aten/src/TH/generic/THTensorMath.cpp:3616