pytorch函数----model.train() 和model.eval()

model.train() :启用 BatchNormalization 和 Dropout

model.eval() :不启用 BatchNormalization 和 Dropout

你可能感兴趣的:(Python)