TypeError

[zhenmafan7]

while I trying to run the train.py, the followed error occured： Traceback (most recent call last): File "/home/wq/Code/DeepGrabCut/train.py", line 147, in loss = class_balanced_cross_entropy_loss(output, gts, size_average=True, batch_average=True) File "/home/wq/Code/DeepGrabCut/layers/loss.py", line 45, in class_balanced_cross_entropy_loss final_loss /= np.prod(label.size()) TypeError: div_() received an invalid combination of arguments - got (numpy.int64), but expected one of:

• (Tensor other) didn't match because some of the arguments have invalid types: (!numpy.int64!)
• (float other) didn't match because some of the arguments have invalid types: (!numpy.int64!)

could you give me some advice to solve this problem?