kd.nn.Dropout

class kauldron.modules.Dropout(

rate: float,

broadcast_dims: collections.abc.Sequence[int] = (),

deterministic: bool | None = None,

rng_collection: str = 'dropout',

parent: flax.linen.module.Module | flax.core.scope.Scope | flax.linen.module._Sentinel | None = <flax.linen.module._Sentinel object>,

name: str | None = None,

)

Bases: flax.linen.stochastic.Dropout

Wrapper around nn.Dropout but using kd.nn.train_property.

property is_training: bool

is_training property.

name: str | None = None

parent: flax.linen.module.Module | flax.core.scope.Scope | flax.linen.module._Sentinel | None = None

scope: Scope | None = None