wumo1556的博客

Ubuntu pytorch 程序退出显存释放问题在多显卡情况下，pytorch程序在退出时显存可能会没有释放。可以通过在终端运行nvidia-smi查看显存的使用。解决方法为，在终端：fusr -v /dev/nvidia* 查看在gpu上运行的所有程序kill 所有连号的进程...

翻译自https://pytorch.org/tutorials/beginner/saving_loading_models.html保存和与加载模型，有三个核心函数需要熟悉：torch.save： 保存一个序列化的对象至硬盘，。该函数使用了Python的pickle包用于序列化。模型、张亮和各种对象的字典都可以使用该函数保存；torch.load：使用pickle的反序列化功能将序列化...

caffe2与PyTorch的图像预处理不同。caffe2中图像的通道与OpenCV相同，是BGR格式，标准化到0-255，PyTorch中图像是RGB格式，标准化到0-1。