Mikhail Sveshnikov
Mikhail Sveshnikov
For now images are not supported, but we are working on this in #310. As for the error, you may try to update to 2.4 since we got some bugs...
Is there a reproducible example in your repo?
They are automatically generated from appropriate model methods. Some examples can be found in tests, but they might not be very clear :( https://github.com/iterative/mlem/blob/ee994a025ab36ce9bb98b4c7960368289c9afe87/tests/contrib/test_torch.py#L127 I will take a look at...
Ok I see now what is going on. You try to save state with mlem, and state is a dict which mlem assumes as data type, not model type. So...
Btw, with fix above saving and loading as dict does work
But you still wont be able to deploy it as only MlemModel can be deployed, not MlemData
Can you check using MLEM from `main` branch?
Hmm, "works on my machine" TM. Can you please also try #397 branch? Btw my env: ``` torch==1.12.1 yolov5==6.2.2 ``` Can you share yours?
Actually, try #453 with `MLEM_DEBUG=true` env. It should print out the module that gives the trouble
It all got merged into `main`, so please re-install from it :) Sorry for the noise and expect a new comment with "now it's in 0.3.0 release, try with `pip...