Update relation_extract.py

bojone · Jul 7, 2019 · ee21deb · ee21deb
1 parent 8e1da6b
commit ee21deb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/relation_extract.py b/relation_extract.py
@@ -211,7 +211,7 @@ def seq_gather(x):
     return K.tf.gather_nd(seq, idxs)
 
 
-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