Skip to content

Spandan-Madan/Pytorch_fine_tuning_Tutorial

Repository files navigation

mit-logo

alt_tag

Pytorch Tutorial for Fine Tuning/Transfer Learning a Resnet for Image Classification

If you want to do image classification by fine tuning a pretrained mdoel, this is a tutorial will help you out. It shows how to perform fine tuning or transfer learning in PyTorch with your own data. It is based on a bunch of of official pytorch tutorials/examples. I felt that it was not exactly super trivial to perform in PyTorch, and so I thought I'd release my code as a tutorial which I wrote originally for my research.

Highly encourage you to run this on a new data set (read main_fine_tuning.py to know which format to store your data in), but for a sample dataset to start with, you can download a simple 2 class dataset from here - https://download.pytorch.org/tutorial/hymenoptera_data.zip

All Torch and PyTorch specific details have been explained in detail in the file main_fine_tuning.py

Hope this tutorial helps you out! :)

Credits - This tutorial is built on top of mainly on 2 Pytorch tutorials - http://pytorch.org/tutorials/beginner/transfer_learning_tutorial.html and https://github.com/pytorch/examples/tree/master/imagenet.

About

A short tutorial on performing fine tuning or transfer learning in PyTorch.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

pFad - Phonifier reborn

Pfad - The Proxy pFad of © 2024 Garber Painting. All rights reserved.

Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.


Alternative Proxies:

Alternative Proxy

pFad Proxy

pFad v3 Proxy

pFad v4 Proxy