wykykyk

请问要怎么保存整个模型呀？我也遇到到了这个问题，但是我在train.py已经修改成save_weights_only=False，即在./nets/callbacks.py设置为了model.save(..)，而不是model.save_weights(..)，训练之后还是报同样的错误。

Traceback (most recent call last): File "h5_convert_to_pb.py", line 71, in h5_model = keras.models.load_model(model_file) File "C:\Anaconda3\envs\vitis-ai1.2\lib\site-packages\keras\engine\saving.py", line 492, in load_wrapper return load_function(*args, **kwargs) File "C:\Anaconda3\envs\vitis-ai1.2\lib\site-packages\keras\engine\saving.py", line 584, in load_model model =...

谢谢谢谢，还是放弃注意力机制了，全是要用到乘法结构