I see, yes - googlenet.py is used to during training of the network when --arch=googlenet is specified.
And I saw some changing googlenet’s pth file method which can set network architecture and property.
Googlenet.py and googlenet-1378be20.pth can change googlenet architecture. This web(Pytorch迁移学习小技巧 以及 Pytorch小技巧的一些总结 - 台部落) can change network architecture. But I don’t know how to do it step by step.
Hi @andy8902, that is more of general PyTorch question unrelated to Jetson, so I recommend consulting the PyTorch tutorials, documentation, and community to find out more about customizing your own networks in PyTorch.
Sorry,I will ask pytorch problem.