其中包含了如何调用GPU以及迁移学习和网络保存以供后续训练 特别是关于网络保存的代码非常又=有借鉴意义
并且成功将训练好的模型在本地能够加载，但是有一点瑕疵不过无伤大雅，由于yaml文件只保存模型结构,而不包含其权重或配置信息，所有我们可以在本地将model重建并且将其保存yaml即可，结合colab上面的weight信息，便可加载模型 其中 model文件夹里面即为链接下载的模型参数及pb文件
-
Notifications
You must be signed in to change notification settings - Fork 0
zxs1652/zcq
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
No description, website, or topics provided.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published