search

omerferhatt / torch2tflite

PyTorch to TensorFlow Lite converter
MIT License
181 stars 43 forks source link

No module named 'models' #5

Closed jhnhp closed 3 years ago

jhnhp commented 3 years ago

I tried this code to convert my custom trained YOLOv5 model (.pt) It returns this message

2021-05-31 10:26:03.181164: I tensorflow/stream_executor/platform/default/dso_loader.cc:53] Successfully opened dynamic library libcudart.so.11.0 No module named 'models'

How can I fix this?

Linxin-Zhong commented 3 years ago

I got the same problem and don't know how to fix this either

mozeqiu commented 3 years ago

I got the same problem and don't know how to fix this either

I suggest to try the steps separately

thongvhoang commented 3 years ago

I got the same problem and don't know how to fix this either

codebugged commented 3 years ago

We are waiting @omerferhatt

mozeqiu commented 3 years ago

我们在等@omerferhatt

I have successfully converted, I will proceed in separate steps。like pt2onnx.py --> onnx2tf.py ---> tf2tflite.py, Instead of finishing the conversion

omerferhatt commented 3 years ago

Hey everyone, sorry for late respond. I change code a little bit. Please test it out and lemme now if anything is broken.

weinixuehao commented 2 years ago

No module named 'model'