natanielruiz
2017-09-14 ec99c6649af6bdbd3c836f20cdc81170e7045cc8
code/batch_testing.py
@@ -123,9 +123,9 @@
            label_roll = labels[:,2].float()
            pre_yaw, pre_pitch, pre_roll, angles = model(images)
            yaw = angles[args.iter_ref-1][:,0].cpu().data
            pitch = angles[args.iter_ref-1][:,1].cpu().data
            roll = angles[args.iter_ref-1][:,2].cpu().data
            yaw = angles[-1][:,0].cpu().data
            pitch = angles[-1][:,1].cpu().data
            roll = angles[-1][:,2].cpu().data
            # Mean absolute error
            yaw_error += torch.sum(torch.abs(yaw - label_yaw) * 3)