Fork me on GitHub
2018年8月15日
Pytorch 0.3加载0.4模型及其之间版本的变化
摘要： 1. 0.4中使用设备：.to(device) 2. 0.4中删除了Variable，直接tensor就可以 3. with torch.no_grad():的使用代替volatile；弃用volatile，测试中不需要计算梯度的话，用with torch.no_grad(): 4. data改用. 阅读全文
posted @ 2018-08-15 22:03 ranjiewen 阅读(2688) 评论(0) 推荐(0) 编辑