Skip to content

GitLab

Commit b519ea47 authored by Rui Qian's avatar Rui Qian Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 466105921
parent 3eed3e03
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/vision/configs/video_classification.py
View file @ b519ea47
...@@ -14,7 +14,7 @@ ...@@ -14,7 +14,7 @@
"""Video classification configuration definition.""" """Video classification configuration definition."""
import dataclasses import dataclasses
from typing import Optional, Tuple from typing import Optional, Tuple, Union
from official.core import config_definitions as cfg from official.core import config_definitions as cfg
from official.core import exp_factory from official.core import exp_factory
from official.modeling import hyperparams from official.modeling import hyperparams
...@@ -45,7 +45,7 @@ class DataConfig(cfg.DataConfig): ...@@ -45,7 +45,7 @@ class DataConfig(cfg.DataConfig):
one_hot: bool = True one_hot: bool = True
shuffle_buffer_size: int = 64 shuffle_buffer_size: int = 64
cache: bool = False cache: bool = False
input_path: str = '' input_path: Union[str, cfg.base_config.Config] = ''
is_training: bool = True is_training: bool = True
cycle_length: int = 10 cycle_length: int = 10
drop_remainder: bool = True drop_remainder: bool = True
......
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