issue with dropout for the inference
thiravet opened this issue · 1 comments
thiravet commented
Hi,
it's a nice and interesting work. But I have a question.
It seems the dropout probability stays at 0.5 for different layers (default value) for the inference with DEMO option (and not only for the training) . Is is normal or did I make a mistake?
Thanks in advance for the answer.
thiravet commented
sorry i didn't have the use of pytorch , i forgot to apply eval() to have the model in inference mode.