分类：保存模型

2021
03-05

解决pytorch 保存模型遇到的问题

今天用pytorch保存模型时遇到bugCan'tpickle<class'torch._C._VariableFunctions'>在google上查找原因，发现是保存时保存了整个模型的原因，而模型中有一些自定义的参数将torch.save(model,save_path)改为torch.save(model.state_dict(),save_path)然后载入模型也做相应的更改就好了补充：pytorch训练模型的一些坑1.图像读取opencv的python和c++读取的图像结果不一致，是因为python和c++采用的opencv版本不一样，从而使用的... 继续阅读 >

2020
10-08

Kears 使用:通过回调函数保存最佳准确率下的模型操作

1：首先，我给我的MixTest文件夹里面分好了类的图片进行重命名（因为分类的时候没有注意导致命名有点不好）defload_data(path):Renamethepicture[atool]foreachoneinos.listdir(path):newname=eachone[7:]os.rename(path+"\\"+eachone,path+"\\"+newname)但是需要注意的是：我们按照类重命名了以后，系统其实会按照图片来排序。这个时候你会看到同一个类的被排序在了一块。这个时候你不要慌张，其实这个顺序是完全... 继续阅读 >

2020
09-29

浅谈keras保存模型中的save()和save_weights()区别

今天做了一个关于keras保存模型的实验，希望有助于大家了解keras保存模型的区别。我们知道keras的模型一般保存为后缀名为h5的文件，比如final_model.h5。同样是h5文件用save()和save_weight()保存效果是不一样的。我们用宇宙最通用的数据集MNIST来做这个实验，首先设计一个两层全连接网络：inputs=Input(shape=(784,))x=Dense(64,activation='relu')(inputs)x=Dense(64,activation='relu')(x)y=Dense(10,activation='s... 继续阅读 >

2020
09-29

解决在keras中使用model.save()函数保存模型失败的问题

我使用的是anaconda安装的环境，其中有一个是h5py，自动安装的是2.7.0的版本，这个版本会导致保存模型时python奔溃。condainstallh5py==2.8.0将h5py更新一个版本即可解决这个问题。补充知识：Django安装提示错误使用sudopipinstall......的时候出现下面一段代码：Thedirectory'/Users/XX/Library/Caches/pip'oritsparentdirectoryisnotownedbythecurrentuserandcachingwheelshasbeendisabled.checkthep... 继续阅读 >