前面我们介绍了基于手写AlexNet以及ResNet18网络得图像分类模型得训练以及相应得TensorRT部署，但是有时候我们想使用其他的主干模型，所以这次采用了Shufflenet作为主干网络进行图像分类的训练。
TensorRT+图像分类：手动搭建AlexNet得图像分类---训练_zhangdaoliang1的博客-CSDN博客
https://blog.csdn.net/zhangdaoliang1/article/details/122982620TensorRT+图像分类：手动搭建AlexNet得图像分类---部署_汤姆&的博客-CSDN博客