NTS-Net icon indicating copy to clipboard operation
NTS-Net copied to clipboard

dropout in resnet.py

Open uniquezhengjie opened this issue 5 years ago • 2 comments

thinks for your great work, I found a err in resnet.py #line 148

x = nn.Dropout(p=0.5)(x)

set net.eval() doesn't work for it ,i think need to defind a self.drop with nn.Dropout(p=0.5) in init function or change to F.dropout(x,p=0.5,training=self.training)

uniquezhengjie avatar Feb 13 '19 03:02 uniquezhengjie

I am facing the same issue. @uniquezhengjie does your solution work?

paramansh avatar Feb 26 '19 05:02 paramansh

This should be fixed since the model returns at test time different results! Took me an hour to figure it out. Should have checked the issues before :-/

DiKorsch avatar Mar 18 '20 14:03 DiKorsch