jupyter notebook加载DDP预训练模型

最近遇到了一个问题，模型是用DistributedDataParallel一机多卡分布式训练的，然后作为一个jupyter notebook重度用户，a56爆大奖在线娱乐想用它来加载这个模型，搞点预测的例子可视化看看。 但是这会碰到一个问题，a56爆大奖在线娱乐们都知道通常加载预训练模型的方法是： pretrained_dict =
posted @ 2021-07-02 10:56  嶙羽  阅读(976)  评论(0)  编辑  收藏  举报