1、TensorFlow和keras
https://robertlexis.github.io/2018/08/28/Tensorflow-模型浮点数计算量和参数量统计/
2、pytorch
一个比较好的包,查看pytorch下的网络信息和参数:https://java.ctolib.com/sksq96-pytorch-summary.html
1、TensorFlow和keras
https://robertlexis.github.io/2018/08/28/Tensorflow-模型浮点数计算量和参数量统计/
2、pytorch
一个比较好的包,查看pytorch下的网络信息和参数:https://java.ctolib.com/sksq96-pytorch-summary.html