In Google Collab, I am trying to convert a simple LSTM model to CoreML.
TensorFlow version is 2.4.1 and the latest coremltools installed with :
!pip install --upgrade coremltools
Still I am getting this error:
ValueError: Unable to determine the type of the model, i.e. the source framework. Please provide the value of argument "source", from one of ["tensorflow", "pytorch", "mil"]. Note that model conversion requires the source package that generates the model. Please make sure you have the appropriate version of source package installed. E.g., if you're converting model originally trained with TensorFlow 1.14, make sure you have tensorflow==1.14 installed.
Has anybody done a conversion of LSTM in Collab? I do succeed to convert other non LSTM models easily.
Thanks for any help.
TensorFlow version is 2.4.1 and the latest coremltools installed with :
!pip install --upgrade coremltools
Still I am getting this error:
ValueError: Unable to determine the type of the model, i.e. the source framework. Please provide the value of argument "source", from one of ["tensorflow", "pytorch", "mil"]. Note that model conversion requires the source package that generates the model. Please make sure you have the appropriate version of source package installed. E.g., if you're converting model originally trained with TensorFlow 1.14, make sure you have tensorflow==1.14 installed.
Has anybody done a conversion of LSTM in Collab? I do succeed to convert other non LSTM models easily.
Thanks for any help.