PYNQ: PYTHON PRODUCTIVITY

Training a deep learning model for DPU compilation

Hi,I am now training a ssd mobile net v1 model on my host using pytorch.
I want to move it to ultra96 v2 board pynq using DPU, but i only see the tutorial using keras and tensorflow!

I am wondering can I compile my model written by pytorch or there is some other tutorial for pytorch version?
Any advice would be appreciated!!
Thanks
Tim

Pytorch should be supported in DPU-PYNQ release v1.3.0. You can follow the steps on Vitis Ai to train your model and use DPU-PYNQ to deploy it.