首页
编程
java
php
前端
首页
编程
java
php
前端
pytorch 的 summary 函数
pytorch summary()函数问题
我的net这一层是从pytorch vgg16模型上取下来的,把output_channel 减半了,可是用summary函数跑出来结果还是64
真的不是很理解
这篇文章:
PyTorch summary 和 冻结网络层
也许能够解决你的问题,你可以看下
点击展开全文