我正在使用 Keras,我很难知道有多少参数Resnet-50。Keras 文档说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
我正在使用 Keras,我很难知道有多少参数Resnet-50。Keras 文档说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
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()
这将为您返回参数总数的正确值。