View source on GitHub |
The hyperparameters for training face stylizers.
Inherits From: BaseHParams
mediapipe_model_maker.face_stylizer.HParams(
learning_rate: float = 0.0008,
batch_size: int = 4,
epochs: int = 100,
steps_per_epoch: Optional[int] = None,
class_weights: Optional[Mapping[int, float]] = None,
shuffle: bool = False,
repeat: bool = False,
export_dir: str = tempfile.mkdtemp(),
distribution_strategy: str = 'off',
num_gpus: int = 0,
tpu: str = ''
)
Methods
get_strategy
get_strategy()
__eq__
__eq__(
other
)