2022年8月27日
Pytorch中获取模型摘要的3种方法
摘要： 在pytorch中获取模型的可训练和不可训练的参数，层名称，内核大小和数量。 Pytorch nn.Module 类中没有提供像与Keras那样的可以计算模型中可训练和不可训练的参数的数量并显示模型摘要的方法 。所以在这篇文章中，我将总结我知道三种方法来计算Pytorch模型中可训练和不可训练的参数 阅读全文
posted @ 2022-08-27 10:08 deephub 阅读(30) 评论(0) 推荐(0) 编辑