I will train a deep learning image classification model in pytorch

I'm a deep learning researcher with published work on image classification and self-supervised learning. I will train a custom PyTorch model (CNN, ResNet, EfficientNet or Vision Transformer) on your image dataset. You get clean, commented code, the trained model file, and accuracy results with charts. Training runs on GPU, so it's fast. Please message me before ordering with your dataset size and goal.