Hi, I would like to know how you guys deal with the dropout in testing since the dropout rate should be set to 0 while testing. I directly set model.training = False while testing. Or is there other ways to handle this?
2 Likes
model.train() for training
model.eval() for testing
8 Likes
That’s cool. Thanks.
what do you mean for " the dropout rate should be set to 0 while testing"?
Is the dropout rate different between train and test?
Yes, it’s. You have to set to 0 when you evaluate your model.
1 Like