lcskrishna / fruit-classification-pytorch

Fruit classification Kaggle dataset using pytorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

[Feature] Apply Transfer learning using existing torchvision networks.

lcskrishna opened this issue · comments

Use existing models such as vgg16 or vgg19 or squeezenet and perform transfer learning.
The goal of this task is to evaluate various models of torchvision and find the accuracy over the validation dataset.

Todo tasks :

  • vgg16 and vgg19
  • resnet50 and resnet101
  • squeezenet
  • densenet