mediapipe_model_maker.gesture_recognizer.HParams

The hyperparameters for training gesture recognizer.

Inherits From: HParams

mediapipe_model_maker.gesture_recognizer.HParams(
    learning_rate: float = 0.001,
    batch_size: int = 2,
    epochs: int = 10,
    steps_per_epoch: Optional[int] = None,
    shuffle: bool = False,
    export_dir: str = tempfile.mkdtemp(),
    distribution_strategy: str = 'off',
    num_gpus: int = -1,
    tpu: str = '',
    lr_decay: float = 0.99,
    gamma: int = 2
)

Attributes
learning_rate	Learning rate to use for gradient descent training.
batch_size	Batch size for training.
epochs	Number of training iterations over the dataset.
lr_decay	Learning rate decay to use for gradient descent training.
gamma	Gamma parameter for focal loss.
steps_per_epoch	Dataclass field
shuffle	Dataclass field
export_dir	Dataclass field
distribution_strategy	Dataclass field
num_gpus	Dataclass field
tpu	Dataclass field

Methods

__eq__

__eq__(
    other
)

Class Variables
batch_size	2
distribution_strategy	'off'
epochs	10
export_dir	'/tmpfs/tmp/tmpdhi734hb'
gamma	2
learning_rate	0.001
lr_decay	0.99
num_gpus	-1
shuffle	False
steps_per_epoch	None
tpu	''