NTS-Net
NTS-Net copied to clipboard
dropout in resnet.py
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)
I am facing the same issue. @uniquezhengjie does your solution work?
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 :-/