abir-hr196 commited on
Commit
e50ec68
·
verified ·
1 Parent(s): 0322b3e

Add README for seed 10

Browse files
Files changed (1) hide show
  1. seed_10/README.md +30 -0
seed_10/README.md ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # MNIST CNN Model (Seed 10)
2
+
3
+ This is a Convolutional Neural Network (CNN) trained on the MNIST dataset with random seed 10.
4
+
5
+ ## Run Details
6
+ - Repository: [abir-hr196/cnn_transfer](https://huggingface.co/abir-hr196/cnn_transfer)
7
+ - Seed-Specific Run: [seed_10](https://huggingface.co/abir-hr196/cnn_transfer/tree/main/seed_10)
8
+ - WandB Run: [run-seed-10](https://wandb.ai/abirharrasse/mnist-cnn-multi-seed/runs/gq53x40d) if current_wandb_run.url else "N/A"
9
+
10
+ ## Model Performance
11
+ - Best Validation Accuracy: 99.17%
12
+
13
+ ## Configuration
14
+ - Architecture: CNN
15
+ - Dataset: MNIST
16
+ - Epochs: 10
17
+ - Batch Size: 128
18
+ - Learning Rate: 0.0003
19
+ - Random Seed Used: 10
20
+
21
+ ## How to Use
22
+
23
+ ```python
24
+ import torch
25
+ # from mnist_transformer import MNISTCNN # Assuming MNISTCNN class definition
26
+
27
+ # model = MNISTCNN()
28
+ # model.load_state_dict(torch.hub.load_state_dict_from_url("https://huggingface.co/abir-hr196/cnn_transfer/resolve/main/seed_10/pytorch_model.bin"))
29
+ # model.eval()
30
+ ```