Spaces:
Sleeping
Sleeping
dazzleun-7
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -32,6 +32,13 @@ else:
|
|
32 |
device = torch.device("cpu")
|
33 |
print('No GPU available, using the CPU instead.')
|
34 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
35 |
|
36 |
class BERTSentenceTransform:
|
37 |
r"""BERT style data transformation.
|
|
|
32 |
device = torch.device("cpu")
|
33 |
print('No GPU available, using the CPU instead.')
|
34 |
|
35 |
+
max_len = 64
|
36 |
+
batch_size = 32
|
37 |
+
warmup_ratio = 0.1
|
38 |
+
num_epochs = 5
|
39 |
+
max_grad_norm = 1
|
40 |
+
log_interval = 200
|
41 |
+
learning_rate = 1e-5
|
42 |
|
43 |
class BERTSentenceTransform:
|
44 |
r"""BERT style data transformation.
|