Resnet-50 中的参数数量

数据挖掘 深度学习 张量流 Python 喀拉斯
2022-02-19 15:14:15

我正在使用 Keras,我很难知道有多少参数Resnet-50Keras 文档25M,如果我model.param_count()在加载Resnet-50模型时使用,它会说234M. 哪一个是正确的?我很困惑。

model.summary打印这个:

Total params: 234,355,586
Trainable params: 210,767,874
Non-trainable params: 23,587,712
2个回答
from keras.applications.resnet50 import ResNet50

resnet_model = ResNet50(weights='imagenet')

#resnet_model.count_params()
resnet_model.summary()

总参数:25,636,712 可训练参数:
25,583,592
不可训练参数:53,120

检查您的代码一次以确保它是 ResNet50

称呼model_name.summary()

这将为您返回参数总数的正确值。