python - 在 PyTorch 中保存模型的框图
问题描述
Keras 有一个很好的功能,可以将模型架构保存为框图
keras.utils.plot_model(self.model, plot_filepath.as_posix(), show_shapes=True)
这可以在 PyTorch 中实现吗?
这个答案谈到了类似于 Keras 的打印摘要。有没有类似保存模型架构的框图?
解决方案
推荐阅读
- javascript - Typescript - 省略 HoC 的 props 类型
- c# - 为什么编辑器窗口不在屏幕中央?
- github - GitHub OAuth2 和 Swagger UI
- animation - 为什么这个旋转动画是垂直移动的?
- sql - 如何从具有重复记录的表中获取唯一事务?
- python - 使用查找结构搜索 pyspark 数据帧
- github - 是否可以使用 GitHub API 更改私有仓库的可见性?
- python - 将数据从自定义 PyQt 小部件传递到主 python 文件
- javascript - 调用withing本身的函数
- javascript - javascript 计数器脚本只是跳到 0