error in exporting model

[Ashbajawed]

any solution for the following error

[GValiente]

Could you post the model you are trying to export?

[Ashbajawed]

Could you post the model you are trying to export?

sure here are the basic checkpoint and model file
https://drive.google.com/file/d/1K6oDZa-wY7NAETi9fM2k_M7YYr6GWocd/view?usp=sharing

https://drive.google.com/file/d/1fuRGGQqsZEMvao7REpYcg95OONWJFd6U/view?usp=sharing

make_model.py (file generating this error ) consist of following code

from network_withoutSE import model
model= model((18,8,8), 1968 ,8)
model.load_weights("b1_episode4001_noSE.h5")

#save model
from pt import export_model
export_model(model, 'pocket_tensor.model')