AttributeError: 'PReLU' object has no attribute 'total_params'
时间: 2024-06-05 10:05:43 浏览: 253
【Python】AttributeError: ‘AutoSchema’ object has no attribute ‘get_link’
5星 · 资源好评率100%
AttributeError: 'PReLU' object has no attribute 'total_params'通常出现在使用Keras时计算模型参数数量时。这个错误的原因是PReLU层的实现中没有定义total_params属性,而该属性是在计算模型参数数量时需要调用的。
如果你想计算Keras模型的参数数量,可以使用下面的代码:
```python
from keras.models import Model
import numpy as np
def count_params(model):
trainable_count = np.sum([K.count_params(w) for w in model.trainable_weights])
non_trainable_count = np.sum([K.count_params(w) for w in model.non_trainable_weights])
return {'trainable': trainable_count, 'non_trainable': non_trainable_count}
model = Model(...)
params = count_params(model)
print(params)
```
这里的Model需要替换为你自己的模型,params将输出模型的可训练参数和不可训练参数的数量。
阅读全文