Open andraspalffy opened 3 years ago
tools/train_and_eval.py, Line, function eval_one_epoch starts with setting model.train(), which is incorrect in my opinion. It should be model.eval().
tools/train_and_eval.py
eval_one_epoch
model.train()
model.eval()
I also found it.
tools/train_and_eval.py
, Line, functioneval_one_epoch
starts with settingmodel.train()
, which is incorrect in my opinion. It should bemodel.eval()
.