KerasTrainerOptions.md

runner.KerasTrainerOptions

View source on GitHub

Provides Keras training related options.

runner.KerasTrainerOptions(
    policy: Optional[Union[str, tf.keras.mixed_precision.Policy]] = None,
    soft_device_placement: bool = False,
    enable_check_numerics: bool = False
)

Attributes
policy	Dataclass field
soft_device_placement	Dataclass field
enable_check_numerics	Dataclass field

Methods

__eq__

__eq__(
    other
)

Return self==value.

Class Variables
enable_check_numerics	False
policy	None
soft_device_placement	False