Upload app.py
Browse files
app.py
CHANGED
|
@@ -18,7 +18,6 @@ model_name = "roberta-base"
|
|
| 18 |
learning_rate = 5e-5
|
| 19 |
batch_size_train = 64
|
| 20 |
step = 1900
|
| 21 |
-
login(token='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')
|
| 22 |
|
| 23 |
file_name = "model-{}.pt".format(step)
|
| 24 |
state_dict = torch.load(os.path.join(file_name))
|
|
|
|
| 18 |
learning_rate = 5e-5
|
| 19 |
batch_size_train = 64
|
| 20 |
step = 1900
|
|
|
|
| 21 |
|
| 22 |
file_name = "model-{}.pt".format(step)
|
| 23 |
state_dict = torch.load(os.path.join(file_name))
|