Google Colabを用いてPytorchのモデルをTFLiteに変換 - Qiita何の記事？ PytorchのモデルからTensorflow Liteへの変換を、Google colabを用いて行う方法です。 Pytorch → ONNX import torch import torchvision.models as models model_name = "resnet50" dummy_input = torch.randn((1, 3, 224, 224)) model = models.resnet50() torch.onnx.export(model, dummy_input, "converted.onnx", verbose=True, input_names=['input'], output_names=['output']) !pip install onnx_tf from onnx_tf.backend import prepare imp
