pytorch-classification icon indicating copy to clipboard operation
pytorch-classification copied to clipboard

Training of image classification models with PyTorch

Results 4 pytorch-classification issues
Sort by recently updated
recently updated
newest added

Hi Dusty, I'm using a **Jetson Nano 2GB**, and using the classification pipeline. I was struggling with the accuracy and confidence level for quite some time. I'm trying to classify...

Hello, I am trying to retrain classification model (mobilenet_v1 or mobilenet_v2), and I am following instructions at : https://github.com/dusty-nv/jetson-inference/blob/master/docs/pytorch-cat-dog.md In Help of train.py it shows architecture as below ``` --model-dir...

Argument 'arch' passed to function reshape_model, but 'args.arch' is used to call it in the last 'else' clause.

Ran into this error when attempting to run training for wide_resnet50_2 as part of UC Berkeley MIDS program W251 class. Seems like a simple copy/paste accident or something, once I...