tensorflow加载模型出现Unable to restore custom object of type _tf_keras_rnn_layer currently.
我在使用自定义层TrajGRU时,先根据ConvLSTMCell重写了TrajGRUCell,然后TrajGRU继承自ConvRNN。写了get_config from_config方法结果还是出现下面报错。
Unable to restore custom object of type _tf_keras_rnn_layer currently.
Please make sure that the layer implements `get_config`and `from_config` when s
原创
2022-05-09 11:22:25 ·
1967 阅读 ·
0 评论