Skip to content

GitLab

Commit 3991a54f authored by Yeqing Li's avatar Yeqing Li Committed by A. Unique TensorFlower
Browse files

Internal changes. 

PiperOrigin-RevId: 288514265
parent a6e44930
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/vision/detection/modeling/architecture/heads.py
View file @ 3991a54f
......@@ -224,7 +224,7 @@ class FastrcnnHead(object):
net = tf.keras.layers.Dense(
units=self._fc_dims,
activation=(None if self._use_batch_norm else tf.nn.relu),
name='fc{}'.format(i+6))(
name='fc{}'.format(i))(
net)
if self._use_batch_norm:
net = self._batch_norm_relu(fused=False)(net, is_training=is_training)
......
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