pocket-tensor
pocket-tensor copied to clipboard
error in exporting model
any solution for the following error
Could you post the model you are trying to export?
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')