1、TensorFlow和keras
https://robertlexis.github.io/2018/08/28/Tensorflow-模型浮点数计算量和参数量统计/
2、pytorch
一个比较好的包,查看pytorch下的网络信息和参数:https://java.ctolib.com/sksq96-pytorch-summary.html
来源:CSDN
作者:庆志的小徒弟
链接:https://blog.csdn.net/qq_27871973/article/details/97642755