2022年4月12日

多GPU模型转换到cpu上
摘要： 3. 多GPU模型转换到cpu上 通过DataParallel包装的model会再加一层module。所以state_dict会多一个module前缀。假设net1 是通过DataParallel包装的模型Net的实例，我们要把它装换到cpu上。方法就是重新建一个对象，把参数迁移过去 state_d 阅读全文
posted @ 2022-04-12 12:21 兔子二 阅读(152) 评论(0) 推荐(0) 编辑

导航