会员
周边
捐助
新闻
博问
闪存
赞助商
Chat2DB
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
嶙羽
博客园
首页
新随笔
联系
订阅
管理
2021年7月2日
jupyter notebook加载DDP预训练模型
摘要: 最近遇到了一个问题,模型是用DistributedDataParallel一机多卡分布式训练的,然后作为一个jupyter notebook重度用户,我想用它来加载这个模型,搞点预测的例子可视化看看。 但是这会碰到一个问题,我们都知道通常加载预训练模型的方法是: pretrained_dict =
阅读全文
posted @ 2021-07-02 10:56 嶙羽
阅读(1054)
评论(0)
推荐(0)
编辑
公告