Fix model card with YAML metadata
Browse files
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
|
| 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()
|