dazzleun-7 commited on
Commit
9829399
·
verified ·
1 Parent(s): 94f9674

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
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.