Renjie WAN

icon indicating an email [email protected]

icon company Hong Kong Baptist University icon location Hong Kong icon location An Assistant Professor from HKBU

Results 1 comments of Renjie WAN

Runtime error: dimension out of range

I also find this problem. To solve this problem, I change the loss = torch.sum(loss,1), loss = torch.sum(loss,2) to loss = torch.sum(loss) directly. Is it right...