deep-head-pose icon indicating copy to clipboard operation
deep-head-pose copied to clipboard

softmax

Open song6cy opened this issue 3 years ago • 0 comments
trafficstars

code/test_on_video_dlib.py:146: UserWarning: Implicit dimension choice for softmax has been deprecated. Change the call to include dim=X as an argument. yaw_predicted = F.softmax(yaw) code/test_on_video_dlib.py:147: UserWarning: Implicit dimension choice for softmax has been deprecated. Change the call to include dim=X as an argument. pitch_predicted = F.softmax(pitch) code/test_on_video_dlib.py:148: UserWarning: Implicit dimension choice for softmax has been deprecated. Change the call to include dim=X as an argument. roll_predicted = F.softmax(roll)

dim is 0,1,or 2?

song6cy avatar Jul 19 '22 08:07 song6cy