search

Xilinx / DPU-PYNQ

DPU on PYNQ
Apache License 2.0
199 stars 68 forks source link

DNU xmodel not working for Pytorch and Tensorflow #77

Closed jnbli closed 2 years ago

jnbli commented 2 years ago

Loading Pytorch and Tensorflow xmodels from Model Zoo don't work. Only loading caffe xmodels seem to work.

skalade commented 2 years ago

Can you provide a bit more info? What board are you using, what models or examples?

jnbli commented 2 years ago

I am using Kria KV260 board. The example is from Kria-PYNQ and the notebook is dpu_resnet50.ipynb.

skalade commented 2 years ago

When you say don't work, is there some error you are getting? How are you downloading the models?

If you are downloading the resnet50 models from the Model Zoo on the Vitis AI github master branch, they are most likely compiled using the v2.0 compiler, whereas we currently run on v1.4, so there could be some incompatibilities there. If that's the case I suggest looking at the v1.4 branch of Vitis AI for downloading models.

You can make sure which version of the model you are downloading by looking at the end of the download url, e.g. https://www.xilinx.com/bin/public/openDownload?filename=resnet50_tf2-zcu102_zcu104_kv260-r1.4.0.tar.gz will download the 1.4.0 version of the model.

skalade commented 2 years ago

Since there's been no activity on this in over 60 days, closing out the issue. I assume the problem was the usage of a 2.0 compiled model with a 1.4 runtime.