Spaces:

keras-io
/

bert-semantic-similarity

Runtime error

vumichien commited on Jul 7, 2022

Commit

f11abfb

1 Parent(s): 9de3728

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ class BertSemanticDataGenerator(tf.keras.utils.Sequence):
         self,
         sentence_pairs,
         labels,
-        batch_size=batch_size,
         shuffle=True,
         include_targets=True,
     ):

         self,
         sentence_pairs,
         labels,
+        batch_size=32,
         shuffle=True,
         include_targets=True,
     ):