face.evoLVe icon indicating copy to clipboard operation
face.evoLVe copied to clipboard

UserWarning: Implicit dimension choice for softmax has been deprecated

Open knotgrass opened this issue 4 years ago • 0 comments

face.evoLVe/applications/align/get_nets.py:70: UserWarning: Implicit dimension choice for softmax has been deprecated. Change the call to include dim=X as an argument. a = F.softmax(a)

specifically in this case, how many output neuron number of this model. I using Onet, Pnet and Rnet Btw, i don't see any flatten layer or linear layer in Onet, Pnet or Rnet Thanks you

knotgrass avatar Nov 15 '21 09:11 knotgrass