polarisZhao / PFLD-pytorch

PFLD pytorch Implementation
798 stars 197 forks source link

想请问下为什么运行train时出现这种错误 应该如何修改 #67

Open daiguangq opened 2 years ago

daiguangq commented 2 years ago

weight_angle = torch.sum(1 - torch.cos(angle - euler_angle_gt), axis=1) RuntimeError: The size of tensor a (3) must match the size of tensor b (0) at non-singleton dimension 1

sainisanjay commented 2 years ago

@daiguangq are you able to solve the problem? Seems i am also getting same error