From ee21deb1cfbc8e09252183ef88b71298797e21b4 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=8B=8F=E5=89=91=E6=9E=97=28Jianlin=20Su=29?= Date: Sun, 7 Jul 2019 21:15:14 +0800 Subject: [PATCH] Update relation_extract.py --- relation_extract.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/relation_extract.py b/relation_extract.py index 0d14f64..171b8a1 100644 --- 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