abir-hr196 commited on
Commit
fbf9be9
·
verified ·
1 Parent(s): ccc607a

Add README for seed 249

Browse files
Files changed (1) hide show
  1. seed_249/README.md +30 -0
seed_249/README.md ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # MNIST CNN Model (Seed 249)
2
+
3
+ This is a Convolutional Neural Network (CNN) trained on the MNIST dataset with random seed 249.
4
+
5
+ ## Run Details
6
+ - Repository: [abir-hr196/cnn_transfer](https://huggingface.co/abir-hr196/cnn_transfer)
7
+ - Seed-Specific Run: [seed_249](https://huggingface.co/abir-hr196/cnn_transfer/tree/main/seed_249)
8
+ - WandB Run: [run-seed-249](https://wandb.ai/abirharrasse/mnist-cnn-multi-seed/runs/i32q2x19) if current_wandb_run.url else "N/A"
9
+
10
+ ## Model Performance
11
+ - Best Validation Accuracy: 99.22%
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: 249
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_249/pytorch_model.bin"))
29
+ # model.eval()
30
+ ```