分类：载入模型

2021
08-08

Pytorch distributed 多卡并行载入模型操作

一、Pytorchdistributed多卡并行载入模型这次来介绍下如何载入模型。目前没有找到官方的distribute载入模型的方式，所以采用如下方式。大部分情况下，我们在测试时不需要多卡并行计算。所以，我在测试时只使用单卡。fromcollectionsimportOrderedDictdevice=torch.device("cuda")model=DGCNN(args).to(device)#自己的模型state_dict=torch.load(args.model_path)#存放模型的位置new_state_dict=OrderedDict()f... 继续阅读 >

2020
10-08

解决Keras 中加入lambda层无法正常载入模型问题

刚刚解决了这个问题，现在记录下来问题描述当使用lambda层加入自定义的函数后，训练没有bug，载入保存模型则显示Nonetypehasnoattribute'get'问题解决方法：这个问题是由于缺少config信息导致的。lambda层在载入的时候需要一个函数，当使用自定义函数时，模型无法找到这个函数，也就构建不了。m=load_model(path,custom_objects={"reduce_mean":self.reduce_mean,"slice":self.slice})其中，reduce_mean和slice定义如下de... 继续阅读 >