方式1 保存了网络模型的结构,也保存了网络模型的参数
运行成功后,会出现如下文件
加载模型,
方式2,保存了模型参数
加载模型后打印出的是字典形式
如果要恢复网络结构,新建一个网络结构vgg16,vgg16加载其中的参数,通过字典的形式,
此时用 方式1有一个陷阱
这是保存后,没有问题
但是当加载时, 报错找不到Tudui这个类
需要将定义的类粘贴在上面
或者前面import一下
方式1 保存了网络模型的结构,也保存了网络模型的参数
运行成功后,会出现如下文件
加载模型,
方式2,保存了模型参数
加载模型后打印出的是字典形式
如果要恢复网络结构,新建一个网络结构vgg16,vgg16加载其中的参数,通过字典的形式,
此时用 方式1有一个陷阱
这是保存后,没有问题
但是当加载时, 报错找不到Tudui这个类
需要将定义的类粘贴在上面
或者前面import一下