![](https://img-blog.csdnimg.cn/20201014180756913.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
WillerW
这个作者很懒,什么都没留下…
展开
-
Ubuntu pytorch 程序退出显存释放问题
Ubuntu pytorch 程序退出显存释放问题在多显卡情况下,pytorch程序在退出时显存可能会没有释放。可以通过在终端运行nvidia-smi查看显存的使用。解决方法为,在终端:fusr -v /dev/nvidia* 查看在gpu上运行的所有程序kill 所有连号的进程...原创 2019-03-12 08:46:26 · 2009 阅读 · 2 评论 -
pytorch 模型加载与保存
翻译自https://pytorch.org/tutorials/beginner/saving_loading_models.html保存和与加载模型,有三个核心函数需要熟悉:torch.save: 保存一个序列化的对象至硬盘,。该函数使用了Python的pickle包用于序列化。模型、张亮和各种对象的字典都可以使用该函数保存;torch.load:使用pickle的反序列化功能将序列化...原创 2019-04-07 14:25:19 · 7200 阅读 · 3 评论 -
caffe2 与 PyTorch 数据预处理
caffe2与PyTorch的图像预处理不同。caffe2中图像的通道与OpenCV相同,是BGR格式,标准化到0-255,PyTorch中图像是RGB格式,标准化到0-1。原创 2019-04-08 20:20:48 · 494 阅读 · 0 评论