search

SanghunYun / UDA_pytorch

UDA(Unsupervised Data Augmentation) implemented by pytorch
Apache License 2.0
275 stars 61 forks source link

model.eval() missing? #10

Open matricer opened 4 years ago

matricer commented 4 years ago

It seems that on model.eval() takes effect when evaluating https://github.com/SanghunYun/UDA_pytorch/blob/master/train.py#L116 https://github.com/SanghunYun/UDA_pytorch/blob/master/train.py#L140 Is that a problem?