I want to convert trained model from PyTorch to tflite.
I saved trained model bellow.
torch.save(net.to(“cpu”).state_dict(), ‘mobilenet_v2.pth’)
How to convert model format from PyTorch to tflite?
python 3.5.6
pytorch 1.3.1
torch 1.4.0
torchvision 0.4.2
tensorflow 2.0.0