![](/img/trans.png)
[英]ValueError: Unknown activation function: my_custom_activation_function
[英]ValueError: Unknown activation function:swish_activation
我正在尝试使用 keras load_model() 加载保存权重。
from keras.models import load_model
model=load_model("weights.hdf5")
这是我得到的错误。
ValueError Traceback (most recent call last)
<ipython-input-34-52d6983dfc34> in <module>()
1 from keras.models import load_model
----> 2 model=load_model("weights.hdf5")
14 frames
/usr/local/lib/python3.6/dist-packages/keras/utils/generic_utils.py in deserialize_keras_object(identifier, module_objects, custom_objects, printable_module_name)
165 if fn is None:
166 raise ValueError('Unknown ' + printable_module_name +
--> 167 ':' + function_name)
168 return fn
169 else:
ValueError: Unknown activation function:swish_activation
Keras 中默认不提供Swish
激活。 相反,添加这个:
from keras.utils.generic_utils import get_custom_objects
from keras import backend as K
from keras.layers import Activation
def swish_activation(x):
return (K.sigmoid(x) * x)
get_custom_objects().update({'swish_activation': Activation(swish_activation)})
# model use some custom objects, so before loading saved model
# import module your network was build with
# e.g. import efficientnet.keras / import efficientnet.tfkeras
import efficientnet.tfkeras
from tensorflow.keras.models import load_model
model = load_model('path/to/model.h5')
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.