Update README.md
Browse files
README.md
CHANGED
|
@@ -94,8 +94,8 @@ model = AutoModelForSequenceClassification.from_pretrained(
|
|
| 94 |
).to(device)
|
| 95 |
|
| 96 |
# 2. Performance Optimization (Mandatory for Monarch Speed)
|
| 97 |
-
torch.set_float32_matmul_precision('high')
|
| 98 |
-
model = torch.compile(model, mode="max-autotune")
|
| 99 |
model.eval()
|
| 100 |
|
| 101 |
# 3. Load MNLI Validation Set
|
|
|
|
| 94 |
).to(device)
|
| 95 |
|
| 96 |
# 2. Performance Optimization (Mandatory for Monarch Speed)
|
| 97 |
+
# torch.set_float32_matmul_precision('high')
|
| 98 |
+
# model = torch.compile(model, mode="max-autotune")
|
| 99 |
model.eval()
|
| 100 |
|
| 101 |
# 3. Load MNLI Validation Set
|