Update subject_extract.py

bojone · Jul 7, 2019 · 992ded0 · 992ded0
1 parent fb9c2a1
commit 992ded0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/subject_extract.py b/subject_extract.py
@@ -147,7 +147,7 @@ def __iter__(self):
 from keras.optimizers import Adam
 
 
-bert_model = load_trained_model_from_checkpoint(config_path, checkpoint_path)
+bert_model = load_trained_model_from_checkpoint(config_path, checkpoint_path, seq_len=None)
 
 for l in bert_model.layers:
     l.trainable = True