Jetson-inference python file calling only the custom dataset model

Hi, I’m using Jetson-Inference. I followed the cat_dog guide on how to train a custom model - the results are great! Now I’m trying to make the python file, but can’t call my specific custom model within resnet-18. The --network tag can only select the base networks, not my custom model. How do I set this up so it only uses my custom-trained model?

Jetson-inference: GitHub - dusty-nv/jetson-inference: Hello AI World guide to deploying deep-learning inference networks and deep vision primitives with TensorRT and NVIDIA Jetson.
cat_dog: jetson-inference/pytorch-cat-dog.md at master · dusty-nv/jetson-inference · GitHub
python file: jetson-inference/imagenet-example-python-2.md at master · dusty-nv/jetson-inference · GitHub