Skip to content

GitLab

Commit 59e25c9d authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 358870598
parent 46c9bf5e
Show whitespace changes
Inline Side-by-side
Showing with 8 additions and 0 deletions
official/vision/beta/configs/common.py
View file @ 59e25c9d
...@@ -27,3 +27,11 @@ class NormActivation(hyperparams.Config): ...@@ -27,3 +27,11 @@ class NormActivation(hyperparams.Config):
use_sync_bn: bool = True use_sync_bn: bool = True
norm_momentum: float = 0.99 norm_momentum: float = 0.99
norm_epsilon: float = 0.001 norm_epsilon: float = 0.001
@dataclasses.dataclass
class PseudoLabelDataConfig(hyperparams.Config):
"""Psuedo Label input config for training."""
input_path: str = ''
data_ratio: float = 1.0 # Per-batch ratio of pseudo-labeled to labeled data
file_type: str = 'tfrecord'
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