Skip to content

GitLab

Commit 0fab1bf1 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 4bfdd769
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/projects/movinet/modeling/movinet_layers.py
View file @ 0fab1bf1
...@@ -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