DropPath
A droppath layer that wraps a given Component.
DropPath(nn.Linear(128, 256), p=0.2)Can be understood as stochastic depth per sample.
Forward
(x: jaxtyping.Float[Tensor, ' *shape']) -> jaxtyping.Float[Tensor, ' *shape']A droppath layer that wraps a given Component.
DropPath(nn.Linear(128, 256), p=0.2)Can be understood as stochastic depth per sample.
(x: jaxtyping.Float[Tensor, ' *shape']) -> jaxtyping.Float[Tensor, ' *shape']