model.train() :啟用 BatchNormalization 和 Dropout
model.eval() :不啟用 BatchNormalization 和 Dropout
參考:
https://pytorch.org/docs/stable/nn.html?highlight=module%20eval#torch.nn.Module.eval
model.train() :啟用 BatchNormalization 和 Dropout
model.eval() :不啟用 BatchNormalization 和 Dropout
參考:
https://pytorch.org/docs/stable/nn.html?highlight=module%20eval#torch.nn.Module.eval
本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。