# I want to train a model with CNN in jetson nano

**URL:** <https://forums.developer.nvidia.com/t/i-want-to-train-a-model-with-cnn-in-jetson-nano/111594>\
**Category:** Jetson Nano\
**Created:** [February 6, 2020, 7:15pm UTC](https://forums.developer.nvidia.com/t/i-want-to-train-a-model-with-cnn-in-jetson-nano/111594 "2020-02-06T19:15:56Z")\
**Posts on this page:** 6\
**Page:** 1

<div class="post-metadata">

**Author:** ![plaketas1](https://developer.download.nvidia.com/images/forums/profile-default-devtalk-84.png) [@plaketas1](https://forums.developer.nvidia.com/u/plaketas1)\
**Post date:** [February 6, 2020, 7:15pm UTC](https://forums.developer.nvidia.com/t/i-want-to-train-a-model-with-cnn-in-jetson-nano/111594/1 "2020-02-06T19:15:56Z")

</div>

Hello,

I want to get some metrics by running a CNN on jetson nano. Has anyone run a Convolution Neural Network from a github or any other source that is reliable and runs for over an hour, so that I can get some metrics about time etc? I have tried some but they do not run in embedded jetson nano + Ubuntu OS etc. That is I ask for your help… Thank you!

---

<div class="post-metadata">

**Author:** ![dusty\_nv](https://sea2.discourse-cdn.com/nvidia/user_avatar/forums.developer.nvidia.com/dusty_nv/32/14043_2.png) [@dusty\_nv](https://forums.developer.nvidia.com/u/dusty_nv)\
**Post date:** [February 6, 2020, 9:30pm UTC](https://forums.developer.nvidia.com/t/i-want-to-train-a-model-with-cnn-in-jetson-nano/111594/2 "2020-02-06T21:30:27Z")

</div>

Hi plaketas1, check out this repo: [https://github.com/dusty-nv/jetson-inference](https://github.com/dusty-nv/jetson-inference)

You could run camera inferencing on one of the included pre-trained models for a long time.

Alternatively, if you are interested in training, it includes transfer learning with PyTorch here:  
[https://github.com/dusty-nv/jetson-inference/blob/master/docs/pytorch-transfer-learning.md](https://github.com/dusty-nv/jetson-inference/blob/master/docs/pytorch-transfer-learning.md)

---

<div class="post-metadata">

**Author:** ![plaketas1](https://developer.download.nvidia.com/images/forums/profile-default-devtalk-84.png) [@plaketas1](https://forums.developer.nvidia.com/u/plaketas1)\
**Post date:** [February 8, 2020, 2:06pm UTC](https://forums.developer.nvidia.com/t/i-want-to-train-a-model-with-cnn-in-jetson-nano/111594/3 "2020-02-08T14:06:38Z")

</div>

In this link:

> <https://github.com/dusty-nv/jetson-inference/blob/master/docs/pytorch-plants.md>

it says:

We’ll use the same training script that we did from the previous example, located under python/training/classification/. By default it’s set to train a ResNet-18 model, but you can change that with the --arch flag.

However the link: python/training/classification/

does not work. How do I do the training?

---

<div class="post-metadata">

**Author:** ![plaketas1](https://developer.download.nvidia.com/images/forums/profile-default-devtalk-84.png) [@plaketas1](https://forums.developer.nvidia.com/u/plaketas1)\
**Post date:** [February 8, 2020, 2:23pm UTC](https://forums.developer.nvidia.com/t/i-want-to-train-a-model-with-cnn-in-jetson-nano/111594/4 "2020-02-08T14:23:07Z")

</div>

I cannot find the

```auto
train.py

```

on folder:

```auto
/Desktop/jetson-inference-master/python/training/classification

```

What is wrong??

---

<div class="post-metadata">

**Author:** ![dusty\_nv](https://sea2.discourse-cdn.com/nvidia/user_avatar/forums.developer.nvidia.com/dusty_nv/32/14043_2.png) [@dusty\_nv](https://forums.developer.nvidia.com/u/dusty_nv)\
**Post date:** [February 21, 2020, 8:03pm UTC](https://forums.developer.nvidia.com/t/i-want-to-train-a-model-with-cnn-in-jetson-nano/111594/5 "2020-02-21T20:03:24Z")

</div>

Do you see anything under that folder, or is it empty?

If it’s empty, the submodules may not have been cloned - try running this:

```auto
$ cd /Desktop/jetson-inference-master
$ git submodule update --init

```

If all else fails, the code for PyTorch classification training is in this submodule: [https://github.com/dusty-nv/pytorch-classification/tree/6b8fcd38fee76cae26e43b9bd547491813bf423d](https://github.com/dusty-nv/pytorch-classification/tree/6b8fcd38fee76cae26e43b9bd547491813bf423d)

---

<div class="post-metadata">

**Author:** ![nadeemm](https://sea2.discourse-cdn.com/nvidia/user_avatar/forums.developer.nvidia.com/nadeemm/32/367020_2.png) [@nadeemm](https://forums.developer.nvidia.com/u/nadeemm)\
**Post date:** [October 14, 2021, 11:54pm UTC](https://forums.developer.nvidia.com/t/i-want-to-train-a-model-with-cnn-in-jetson-nano/111594/6 "2021-10-14T23:54:23Z")

</div>