PyTorch保存模型、冻结参数等
摘要：
此外可以参考PyTorch模型保存。https://zhuanlan.zhihu.com/p/73893187 查看模型每层输出详情 Keras有一个简洁的API来查看模型的每一层输出尺寸，这在调试网络时非常有用。现在在PyTorch中也可以实现这个功能。 使用很简单，如下用法： input_siz 阅读全文
posted @ 2022-04-19 15:28 lmqljt 阅读(1322) 评论(0) 推荐(0) 编辑
