# Training a deep learning model for DPU compilation

**URL:** <https://discuss.pynq.io/t/training-a-deep-learning-model-for-dpu-compilation/2357>\
**Category:** Support\
**Created:** [February 24, 2021, 6:13am UTC](https://discuss.pynq.io/t/training-a-deep-learning-model-for-dpu-compilation/2357 "2021-02-24T06:13:25Z")\
**Posts on this page:** 2\
**Page:** 1

<div class="post-metadata">

**Author:** ![TingShen\_Kuo](https://sea1.discourse-cdn.com/flex019/user_avatar/discuss.pynq.io/tingshen_kuo/32/914_2.png) [@TingShen\_Kuo](https://discuss.pynq.io/u/TingShen_Kuo)\
**Post date:** [February 24, 2021, 6:13am UTC](https://discuss.pynq.io/t/training-a-deep-learning-model-for-dpu-compilation/2357/1 "2021-02-24T06:13:25Z")

</div>

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!

> <https://github.com/Xilinx/DPU-PYNQ/blob/master/host/train_mnist_model.ipynb>

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

---

<div class="post-metadata">

**Author:** ![rock](https://sea1.discourse-cdn.com/flex019/user_avatar/discuss.pynq.io/rock/32/32_2.png) [@rock](https://discuss.pynq.io/u/rock)\
**Post date:** [March 5, 2021, 4:59pm UTC](https://discuss.pynq.io/t/training-a-deep-learning-model-for-dpu-compilation/2357/2 "2021-03-05T16:59:44Z")

</div>

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.