pytorch model(torch.nn.Module)笔记
摘要:
参考:『PyTorch』第十四弹_torch.nn.Module类属性 官网:https://ptorch.com/docs/1/torch-nn model.eval() # 将本层及子层的training设定为False 在测试阶段,dropout什么都不做 model.train() # 将本 阅读全文
posted @ 2019-07-09 15:19 cekong 阅读(535) 评论(0) 推荐(1) 编辑