Do model.train and model.eval need to be added？

floodsung / LearningToCompare_FSL

PyTorch code for CVPR 2018 paper: Learning to Compare: Relation Network for Few-Shot Learning (Few-Shot Learning part)

MIT License

1.04k stars 267 forks source link

Open zhantt626 opened 1 year ago

zhantt626 commented 1 year ago

Do model.train and model.eval need to be added during the training?Looking forwards your reply!!