Pytorch 模型查看
def count_parameters(model):
return sum(p.numel() for p in model.parameters() if p.requires_grad)print("模型的参数量", count_parameters(model))
print("model",model)
from torchsummary import summary
print(summary(model, (input_data.shape[1])))
推荐阅读
- Linux下面如何查看tomcat已经使用多少线程
- pytorch|使用pytorch从头实现多层LSTM
- MySql数据库备份与恢复
- Flutter的ListView
- Pytorch|Pytorch AlexNet Fashion-MNIST
- 一般模型化关系——从模型是什么到如何起作用的基本答案
- 如何在手机上查看测试vue-cli构建的项目
- pytorch|YOLOX 阅读笔记
- Pytorch学习|sklearn-SVM 模型保存、交叉验证与网格搜索
- 旅途碎碎念