model.train() :启用 BatchNormalization 和 Dropout
model.eval() :不启用 BatchNormalization 和 Dropout
参考链接:https://pytorch.org/docs/stable/nn.html?highlight=module%20eval#torch.nn.Module.eval
2024-03-08 14:09:28
model.train() :启用 BatchNormalization 和 Dropout
model.eval() :不启用 BatchNormalization 和 Dropout
参考链接:https://pytorch.org/docs/stable/nn.html?highlight=module%20eval#torch.nn.Module.eval
下一篇:dropout原理学习