chandu1617 commited on
Commit
58dc2a6
Β·
verified Β·
1 Parent(s): 3ae70e3

Fix model card with YAML metadata

Browse files
Files changed (1) hide show
  1. README.md +2 -3
README.md CHANGED
@@ -1,4 +1,3 @@
1
- %%writefile README.md
2
  ---
3
  library_name: pytorch
4
  tags:
@@ -13,7 +12,7 @@ metrics:
13
  - accuracy
14
  ---
15
 
16
- # MNIST MLP (fold‑4 best)
17
 
18
  **Model**: `ImprovedMLP` (2048 β†’ 1024 β†’ 512 β†’ 256 β†’ 128 β†’ 10)
19
  **File**: `mlp_best_fold4.pth`
@@ -42,4 +41,4 @@ class ImprovedMLP(nn.Module):
42
  path = hf_hub_download("chandu1617/MNIST_with_MLP", "mlp_best_fold4.pth")
43
  model = ImprovedMLP()
44
  model.load_state_dict(torch.load(path))
45
- model.eval()
 
 
1
  ---
2
  library_name: pytorch
3
  tags:
 
12
  - accuracy
13
  ---
14
 
15
+ # MNIST MLP (fold-4 best)
16
 
17
  **Model**: `ImprovedMLP` (2048 β†’ 1024 β†’ 512 β†’ 256 β†’ 128 β†’ 10)
18
  **File**: `mlp_best_fold4.pth`
 
41
  path = hf_hub_download("chandu1617/MNIST_with_MLP", "mlp_best_fold4.pth")
42
  model = ImprovedMLP()
43
  model.load_state_dict(torch.load(path))
44
+ model.eval()