search

nvdla / sw

NVDLA SW
Other
483 stars 191 forks source link

How to use tensorflow model on VP? #59

Open jimchen2018 opened 6 years ago

jimchen2018 commented 6 years ago

Hi Sirs, The opensource roadmap says ‘Application supporting Caffe and TensorFlow models; most CNN related layers’，but I don't know how to use tensorflow model on VP. Could you show me an example? Thx.

prasshantg commented 6 years ago

@jimchen2018 TensorFlow is not yet supported by DLA compiler. It is in our future plan. Please let us know if it is blocking issue for you. Can you use TF model by converting to Caffe?

jimchen2018 commented 6 years ago

Now I use caffe to train my network.
The blocking issue for me is the NLA compiler. Hope it can support SSD network. Thx.

crypticmystic commented 5 years ago

Can I suggest that a toolset (like mmdnn, but one that is a bit more forgiving) could convert tensorflow -> caffe, and parts that can't be directly converted could be left a custom "to be hand edited" parts? Or does such a tool already exist?

Jwy-jump commented 5 years ago

@jimchen2018 can DLA support SSD?

fisherxue commented 5 years ago

@prasshantg Do you know if it's possible to get TensorRT to work on VP? Because then it would be possible to run Tensorflow graphs using TensorRT. Currently in the process of getting it to work, wondering if you guys have tried it before and what your thoughts are. VP doesn't virtualize a CUDA device as far as I know.