Batch testing and hopenet training complete

natanielruiz

2017-09-13 c495a0f6b13b794bab9f6e3423d5038ce645d816

 code/test_AFLW.py

@@ -97,7 +97,7 @@
        label_pitch = labels[:,1].float()
        label_roll = labels[:,2].float()

        yaw, pitch, roll = model(images)
        yaw, pitch, roll, angles = model(images)

        # Binned predictions
        _, yaw_bpred = torch.max(yaw.data, 1)