model.train() :启用 BatchNormalization 和 Dropout
model.eval() :不启用 BatchNormalization 和 Dropout
pytorch函数----model.train() 和model.eval()
最新推荐文章于 2023-04-05 21:12:27 发布
model.train() :启用 BatchNormalization 和 Dropout
model.eval() :不启用 BatchNormalization 和 Dropout