- This repository is a beginner's guide to Transfer Learning using Pytorch.
- It is a concise implementation of how to generalise on a very small dataset (<250 training images) using something called Transfer Learning.
- I have used a pretrained ResNet model with both finetuning the entire net as well as just adjusting the weights of the final layer to quickly generalise on this dataset and get accuracy of more than 95%.
-
Notifications
You must be signed in to change notification settings - Fork 0
This repository is a beginner's guide to Transfer Learning using Pytorch. It is a concise implementation of how to generalise on a very small dataset (<250 training images) using something called Transfer Learning.
Vortexx2/ants-and-bees
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
This repository is a beginner's guide to Transfer Learning using Pytorch. It is a concise implementation of how to generalise on a very small dataset (<250 training images) using something called Transfer Learning.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published