chandu1617 commited on
Commit
fd7fab0
·
verified ·
1 Parent(s): 0992e78

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +12 -9
README.md CHANGED
@@ -1,10 +1,13 @@
1
- ---
2
- tags:
3
- - model_hub_mixin
4
- - pytorch_model_hub_mixin
5
- ---
6
 
7
- This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
8
- - Code: [More Information Needed]
9
- - Paper: [More Information Needed]
10
- - Docs: [More Information Needed]
 
 
 
 
 
 
 
 
1
+ # CIFAR-10 CNN Model
 
 
 
 
2
 
3
+ This is a convolutional neural network trained on the CIFAR-10 dataset, achieving 92.59% test accuracy after 100 epochs.
4
+ ## Model Details
5
+ - Architecture: 9 convolutional layers with batch normalization, max pooling, and dropout, followed by 3 fully connected layers.
6
+ - Dataset: CIFAR-10 (10 classes: airplane, automobile, bird, cat, deer, dog, frog, horse, ship, truck).
7
+ - Training: 100 epochs, SGD optimizer, CrossEntropyLoss, learning rate scheduling.
8
+ ## Usage
9
+ Load the model using:
10
+ ```python
11
+ from huggingface_hub import from_pretrained_pytorch
12
+ model = from_pretrained_pytorch('chandu1617/CIFAR10-CNN_Model')
13
+ ```