dusty-nv / jetson-inference

Hello AI World guide to deploying deep-learning inference networks and deep vision primitives with TensorRT and NVIDIA Jetson.
https://developer.nvidia.com/embedded/twodaystoademo
MIT License
7.74k stars 2.97k forks source link

MobileNetV3 constant dips in accuracy. #1835

Closed ConfusedPelican80 closed 4 months ago

ConfusedPelican80 commented 4 months ago

Hello. First time opening an issue on GitHub so I apologize if this is lacking something. I am currently using:

Jetson Nano Dev Kit 4 Gb. Python 3.6.9 Pytorch (torch) 1.10.0 TorchVision 0.11 JetPack SDK 4.6.1-b110 Linux4Tegra (L4T) Ubuntu 18.04.5 Docker 20.10.21 r32.7.4 Docker Container

I have been following the transfer learning tutorial and have trained a few models such as EfficientNet-B2, ResNet-18, etc with accuracy ranging from 82-88% on a plant disease classification dataset for my capstone project. However, when I tried to use the MobileNetV3_small model. I noticed that there was a constant decline in accuracy. The accuracy constantly dips and never seems to recover.

Capture