2021年1月28日
pytorch模型与参数的GPU，cpu加载转换
摘要： 我们在使用pytorch的过程，经常会需要加载模型参数，不管是别人提供给我们的模型参数，还是我们自己训练的模型参数，那么加载模型参数就会碰到一些情况，即GPU模型和CPU模型，这两种模型是不能混为一谈的，下面分情况进行操作说明。 情况一：模型是GPU模型，预加载的训练参数也是GPU；模型是CPU模型 阅读全文
posted @ 2021-01-28 19:50 Parallax 阅读(672) 评论(0) 推荐(0) 编辑