model.eval()
时间: 2023-07-14 07:00:05 浏览: 53
pytorch:model.train和model.eval用法及区别详解
model.train()是用于训练模型的函数,会启用训练模式,如果模型有dropout层或者batch normalization层,会在训练模式下起作用。
model.eval()是用于评估模型的函数,会启用评估模式,如果模型有dropout层或者batch normalization层,会在评估模式下不起作用。
阅读全文