Skip to content

GitLab

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):
stochastic_depth_drop_rate: float = 0.2
use_external_states: bool = False
average_pooling_type: str = '3d'
output_states: bool = True
@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