Minor bug fix

PiperOrigin-RevId: 416066044
google-research · Dec 13, 2021 · 932b41f · 932b41f
1 parent c77cfcf
commit 932b41f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/classifier_utils.py b/classifier_utils.py
@@ -779,7 +779,7 @@ def create_model(albert_config, is_training, input_ids, input_mask, segment_ids,
       use_einsum=True,
       hub_module=hub_module)
 
-  hidden_size = output_layer.shape[-1].value
+  hidden_size = output_layer.shape[-1]
 
   output_weights = tf.get_variable(
       "output_weights", [num_labels, hidden_size],