def METHOD_NAME(self):
    torch_model = TestExportModes.MyModel()
    fake_input = Variable(torch.randn(1, 1, 224, 224), requires_grad=True)
    f = io.BytesIO()
    torch.onnx._export(
        torch_model,
        (fake_input),
        f,
        verbose=False,
        export_type=torch.onnx.ExportTypes.PROTOBUF_FILE,
    )