Hugging Face's logo

ChandraPrakashBathula
/
CIFAR10-CNN_Model

Image Classification
Safetensors
PyTorch
English
computer-vision
cifar10
Model card Files
xet
 Community
Chandra Prakash Bathula commited on
Commit
cb8f784
·
verified ·
1 Parent(s): 0a588e1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -7
README.md CHANGED
@@ -2,14 +2,15 @@
2
  language: en
3
  license: mit
4
  tags:
5
- - image-classification
6
- - computer-vision
7
- - pytorch
8
- - cifar10
9
  datasets:
10
- - cifar10
11
  metrics:
12
- - accuracy
 
13
  ---
14
 
15
  # CIFAR-10 CNN Model
@@ -36,4 +37,4 @@ Try the model in an interactive Gradio UI at [chandu1617/cifar10-cnn-demo](https
36
  ## Training Details
37
  - **Optimizer**: SGD with momentum 0.9, weight decay 1e-6.
38
  - **Learning Rate**: Initial 0.01, reduced on plateau (factor 0.1, patience 10, min_lr 0.00001).
39
- - **Data Augmentation**: Color jitter, random perspective, random horizontal flip, normalization.
 
2
  language: en
3
  license: mit
4
  tags:
5
+ - image-classification
6
+ - computer-vision
7
+ - pytorch
8
+ - cifar10
9
  datasets:
10
+ - cifar10
11
  metrics:
12
+ - accuracy
13
+ pipeline_tag: image-classification
14
  ---
15
 
16
  # CIFAR-10 CNN Model
 
37
  ## Training Details
38
  - **Optimizer**: SGD with momentum 0.9, weight decay 1e-6.
39
  - **Learning Rate**: Initial 0.01, reduced on plateau (factor 0.1, patience 10, min_lr 0.00001).
40
+ - **Data Augmentation**: Color jitter, random perspective, random horizontal flip, normalization.