gokul-pv / VisionTransformer Public

Notifications You must be signed in to change notification settings
Fork 0
Star 0

Vision transformer trained for cats vs dogs

0 stars 0 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 6 Commits
CatsVsDogs		CatsVsDogs
ViT		ViT
README.md		README.md

Repository files navigation

ViT - An Image is worth 16x16 words

Part 1: (Click here)

Implemented this blog and trained the ViT model for Cats vs Dogs. Transfer learning can be used if needed.

Part 2 (Click here)

Explanation for all of the Vision Transformer classes.

About

Vision transformer trained for cats vs dogs

pytorch transfer-learning attention-mechanism vision-transformer

Report repository

Releases

No releases published

Packages

No packages published

Languages

Jupyter Notebook 100.0%