Skip to content

GitLab

"...git@developer.sourcefind.cn:OpenDAS/mmdetection3d.git" did not exist on "9c7270d00dbdd0599b6b6bf816c3ff2dd17d4878"
Commit ccd93b61 authored by Liangzhe Yuan's avatar Liangzhe Yuan Committed by A. Unique TensorFlower
Browse files

#movinet Also output state if no pooling is performed in the movinet head. 

PiperOrigin-RevId: 467322142
parent 6963a218
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/projects/movinet/modeling/movinet_layers.py
View file @ ccd93b61
...@@ -1455,7 +1455,7 @@ class Head(tf.keras.layers.Layer): ...@@ -1455,7 +1455,7 @@ class Head(tf.keras.layers.Layer):
if self._pool is not None: if self._pool is not None:
outputs = self._pool(x, states=states, output_states=True) outputs = self._pool(x, states=states, output_states=True)
else: else:
outputs = x outputs = (x, states)
return outputs return outputs
......
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