kevin-nd commited on
Commit
aaa8318
·
verified ·
1 Parent(s): 648e8d7

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +27 -8
README.md CHANGED
@@ -1,13 +1,32 @@
1
  ---
 
 
 
 
 
 
2
  license: mit
3
  pipeline_tag: image-classification
4
  tags:
5
- - arxiv:1512.03385
6
- - model_hub_mixin
7
- - pytorch_model_hub_mixin
8
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
9
 
10
- This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
11
- - Code: https://github.com/K-bNd/cv_101
12
- - Paper: [More Information Needed]
13
- - Docs: [More Information Needed]
 
1
  ---
2
+ {}
3
+ ---
4
+ datasets:
5
+ - imagenet
6
+ language: en
7
+ library_name: pytorch
8
  license: mit
9
  pipeline_tag: image-classification
10
  tags:
11
+ - image-classification
12
+ - pytorch
13
+ - lightning
14
+ - imagenet
15
+
16
+ # ResNet50 trained on ImageNet-1k
17
+
18
+ This model was trained using the [cv_101](https://github.com/K-bNd/cv_101) deep learning framework.
19
+
20
+ - **Task:** image-classification
21
+ - **Architecture:** ResNet50
22
+ - **Dataset:** ImageNet-1k
23
+ - **Framework:** PyTorch + PyTorch Lightning
24
+
25
+ ## Test Results
26
+
27
+ | Metric | Value |
28
+ |--------|-------|
29
+ | Accuracy | 0.7476 |
30
+ | Top-5 Accuracy | 0.9197 |
31
+ | Loss | 2.1267 |
32