2021年7月2日
jupyter notebook加载DDP预训练模型
摘要： 最近遇到了一个问题，模型是用DistributedDataParallel一机多卡分布式训练的，然后作为一个jupyter notebook重度用户，我想用它来加载这个模型，搞点预测的例子可视化看看。 但是这会碰到一个问题，我们都知道通常加载预训练模型的方法是： pretrained_dict = 阅读全文
posted @ 2021-07-02 10:56 嶙羽 阅读(1004) 评论(0) 推荐(0) 编辑