Skip to content

Trying to code Resnet50 on pytorch and testing it on CIFAR10 dataset

Notifications You must be signed in to change notification settings

krag-harsh/Resnet50

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 

Repository files navigation

Resnet50

Coded Resnet50 on pytorch and tested it on CIFAR10 dataset

Achieved 69% accuracy with 2 epoch on CIFAR10 dataset. There are comments to understand the flow Used cross entropy loss with SGD optimizer. Used batch size=128, learning rate = 0.01.

Added focal loss