TFlite conversion not working

[mominali12]

Hi I am trying to convert mcunet-in2 model to tflite and does not work using generate_tflite.py script. Would be grateful if some one can help me identify why it breaks.

Commad used to execute the script:
python generate_tflite.py "./mcunet-256kb-1mb_imagenet.json" "/home/mal/ship_paper/ship_detection/code/mcunet-256kb-1mb_imagenet.pth" "./momin_mcunet_test.tflite"

Error I am getting:
E tensorflow/core/grappler/grappler_item_builder.cc:656] Init node first_conv/conv/weight/Assign doesn't exist in graph

Environment / Libraries version used:

Name: tensorflow
Version: 1.15.0

Name: torch
Version: 1.5.0

Name: torchvision
Version: 0.6.0