Resnet network program on jetson nano

When I used this web(jetson-inference/pytorch-collect.md at master · dusty-nv/jetson-inference · GitHub) 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(jetson-inference/pytorch-collect.md at master · dusty-nv/jetson-inference · GitHub). 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.