SergioMtz commited on
Commit
23bae13
·
1 Parent(s): 56553fa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -17,6 +17,7 @@ START = tf.argmax(tf.constant(reserved_tokens) == "[START]")
17
  END = tf.argmax(tf.constant(reserved_tokens) == "[END]")
18
  PAD = tf.argmax(tf.constant(reserved_tokens) == "[PAD]")
19
  VOCAB_SIZE = len(vocab)
 
20
  D_MODEL = 256
21
  NB_LAYERS = 6
22
  FFN_UNITS = 2048
 
17
  END = tf.argmax(tf.constant(reserved_tokens) == "[END]")
18
  PAD = tf.argmax(tf.constant(reserved_tokens) == "[PAD]")
19
  VOCAB_SIZE = len(vocab)
20
+ print(VOCAB_SIZE)
21
  D_MODEL = 256
22
  NB_LAYERS = 6
23
  FFN_UNITS = 2048