appvoid commited on
Commit
8a5f35c
·
verified ·
1 Parent(s): 2dadb3c

Update bet_model.py

Browse files
Files changed (1) hide show
  1. bet_model.py +1 -1
bet_model.py CHANGED
@@ -7,7 +7,7 @@ import torch.nn.functional as F
7
  from torch.utils.checkpoint import checkpoint
8
 
9
  SEQ_LEN = 1024
10
- MAX_LOOPS = 8
11
  PAD_ID, BOS_ID, EOS_ID = 256, 257, 258
12
  VOCAB_SIZE = 259
13
  EXPECTED_PARAM_COUNT = 9_353_876
 
7
  from torch.utils.checkpoint import checkpoint
8
 
9
  SEQ_LEN = 1024
10
+ MAX_LOOPS = 10
11
  PAD_ID, BOS_ID, EOS_ID = 256, 257, 258
12
  VOCAB_SIZE = 259
13
  EXPECTED_PARAM_COUNT = 9_353_876