Dropout & DropoutV2 Layer
Closed this issue · 1 comments
unanan commented
In my net written in Pytorch, I use a Dropout Layer with 0.25 input probability, so PytorchConverter translate it into a dropout layer in Ncnn with label "DropoutV2". After that I test my net in Ncnn, and it died in this "DropoutV2" Layer.
To ensure the reason, I change the probability in the source codes in Pytorch, that I change probability of the Dropout Layer into 0.5, so that PytorchConverter can translate it into a dropout layer in Ncnn with label "Dropout". After that I test my net in Ncnn, and it went well.
I report this problem to Ncnn's author, and she advised me to open a issue on PytorchConverter.
Hope you can tell me how to deal with the problem.