Resnet network program on jetson nano

When I used this web(https://github.com/dusty-nv/jetson-inference/blob/master/docs/pytorch-collect.md) to train resnet , the network program’s folder is upper picture

.

The web’s training program is bottom picture

.

Hi,

Do you meet any error about this?
The name is different. One is file name and the other should be the class name.

Thanks.

Hello,

I use this web(https://github.com/dusty-nv/jetson-inference/blob/master/docs/pytorch-collect.md). The train.py picture is below. The blue circle is default’s network:Resnet18. Does the network download on terminal?

Hi @andy8902, yes, PyTorch automatically downloads the resnet18 model for you. Then train.py re-trains it on the dataset.