arpit-gour02 commited on
Commit
f67244a
·
verified ·
1 Parent(s): 75ee554

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -1
README.md CHANGED
@@ -50,7 +50,14 @@ It was trained using **Transfer Learning**, starting with weights pre-trained on
50
  - **Model type:** Computer Vision (Image Classification / CNN)
51
  - **Language(s) (NLP):** English (Implicitly, via the text present in the RVL-CDIP dataset images)
52
  - **License:** MIT
53
- - **Finetuned from model:** ResNet-50 (ImageNet weights)
 
 
 
 
 
 
 
54
 
55
  ### Model Sources
56
 
 
50
  - **Model type:** Computer Vision (Image Classification / CNN)
51
  - **Language(s) (NLP):** English (Implicitly, via the text present in the RVL-CDIP dataset images)
52
  - **License:** MIT
53
+
54
+ ## Why ResNet50
55
+
56
+ | Model | Approximate Parameters | Year Released | Layers |
57
+ |------------|------------------------|---------------|--------|
58
+ | VGG16 | 138.4 Million | 2014 | 16 |
59
+ | AlexNet | 61.1 Million | 2012 | 8 |
60
+ | ResNet-50 | 25.6 Million | 2015 | 50 |
61
 
62
  ### Model Sources
63