runner.KerasTrainerOptions
December 14, 2023 ยท View on GitHub
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
|