Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -56,6 +56,8 @@ class DistillBERTClass(torch.nn.Module):
|
|
| 56 |
|
| 57 |
|
| 58 |
model = DistillBERTClass()
|
|
|
|
|
|
|
| 59 |
optimizer = torch.optim.Adam(params = model.parameters(), lr=LEARNING_RATE)
|
| 60 |
checkpoint = torch.load("pytorch_distilbert_news.bin")
|
| 61 |
model.load_state_dict(checkpoint['model'])
|
|
|
|
| 56 |
|
| 57 |
|
| 58 |
model = DistillBERTClass()
|
| 59 |
+
LEARNING_RATE = 1e-05
|
| 60 |
+
|
| 61 |
optimizer = torch.optim.Adam(params = model.parameters(), lr=LEARNING_RATE)
|
| 62 |
checkpoint = torch.load("pytorch_distilbert_news.bin")
|
| 63 |
model.load_state_dict(checkpoint['model'])
|