layers.drop_path_config

Classes

UnquantizedDropPathConfig

Configuration for the UnquantizedDropPath layer.

Module Contents

class layers.drop_path_config.UnquantizedDropPathConfig(/, **data)

Bases: pydantic.BaseModel

Configuration for the UnquantizedDropPath layer.

Create a new model by parsing and validating input data from keyword arguments.

Raises [ValidationError][pydantic_core.ValidationError] if the input data cannot be validated to form a valid model.

self is explicitly positional-only to allow self as a field name.

Parameters:

data (Any)

drop_prob: float = None

Probability of dropping a path during training.

scale_by_keep: bool = None

Up-scales activations during training by 1 - drop_prob to avoid train-test mismatch. Defaults to True.