Skip to content

GitLab

"...git@developer.sourcefind.cn:OpenDAS/mmdetection3d.git" did not exist on "bb204696c19ded90121caea82e34fa3ff0bd131f"
Commit 5ed215b2 authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

Remove set_shape as tfhub module updated. 

PiperOrigin-RevId: 290977517
parent d6117ed1
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 3 deletions
official/nlp/bert_models.py
View file @ 5ed215b2
...@@ -309,9 +309,6 @@ def squad_model(bert_config, ...@@ -309,9 +309,6 @@ def squad_model(bert_config,
core_model = hub.KerasLayer(hub_module_url, trainable=True) core_model = hub.KerasLayer(hub_module_url, trainable=True)
_, sequence_output = core_model( _, sequence_output = core_model(
[input_word_ids, input_mask, input_type_ids]) [input_word_ids, input_mask, input_type_ids])
# Sets the shape manually due to a bug in TF shape inference.
# TODO(hongkuny): remove this once shape inference is correct.
sequence_output.set_shape((None, max_seq_length, bert_config.hidden_size))
squad_logits_layer = BertSquadLogitsLayer( squad_logits_layer = BertSquadLogitsLayer(
initializer=initializer, float_type=float_type, name='squad_logits') initializer=initializer, float_type=float_type, name='squad_logits')
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment