【pytorch】Pytorch中的 model.train和 model.eval模式區別

model.train() :啓用 BatchNormalization層 和 dropout層

model.eval() :不啓用 BatchNormalization層 和 dropout層

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章