How to use other pretrained models in Jetbot?

Hello, I want to try SqueezeNet in jetbot.

In my case, code is road_following.
https://github.com/NVIDIA-AI-IOT/jetbot/blob/master/notebooks/road_following/train_model.ipynb

I think output of SqueezeNet is different from ResoNet18 in train_model.ipynb, so another layer must be added same as Resnet18 in train_model.ipynb.

Thank you for your help,