Skip to content

GitLab

"...hubert/git@developer.sourcefind.cn:OpenDAS/torchaudio.git" did not exist on "184466a904c45d3a7e84629bff647b874f5448d5"
Commit 3fb4415e authored by Liangzhe Yuan's avatar Liangzhe Yuan Committed by A. Unique TensorFlower
Browse files

#movinet Add output_state flag. Default to be True. 

PiperOrigin-RevId: 467711326
parent 0fab1bf1
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
official/projects/movinet/configs/movinet.py
View file @ 3fb4415e
...@@ -54,6 +54,7 @@ class Movinet(hyperparams.Config): ...@@ -54,6 +54,7 @@ class Movinet(hyperparams.Config):
stochastic_depth_drop_rate: float = 0.2 stochastic_depth_drop_rate: float = 0.2
use_external_states: bool = False use_external_states: bool = False
average_pooling_type: str = '3d' average_pooling_type: str = '3d'
output_states: bool = True
@dataclasses.dataclass @dataclasses.dataclass
......
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