mazzzystar/Queryable

pytorch version and coremltools version？export failed with coremltools.

andforce opened this issue · 2 comments

andforce commented

Hello,
I encountered a failure when exporting with coremltools:

Converting PyTorch Frontend ==> MIL Ops: 6%|▎ | 63/1131 [00:00<00:00, 10698.02 ops/s]
...
RuntimeError: PyTorch convert function for op 'unflatten' not implemented.

which versions of PyTorch and coremltools you are using?

My development environment:
m1 mac pro
python: 3.8.17
torch: 2.0.0
coremltools: 6.3.0

Thanks~

mazzzystar commented

I think the major difference is PyTorch.

Python 3.9.15
>>> print(torch.__version__)
1.13.1
>>> print(coremltools.__version__)
7.0b1
  • 👍1
andforce commented

I think the major difference is PyTorch.

Python 3.9.15
>>> print(torch.__version__)
1.13.1
>>> print(coremltools.__version__)
7.0b1

thanks,
it works when i downgrade PyTorch to 1.13.1

  • 👍1