(self)
| 158 | pscan: bool = True # use parallel scan mode or sequential mode when training |
| 159 | |
| 160 | def __post_init__(self): |
| 161 | self.d_inner = self.expand_factor * self.dim # E*D = ED in comments |
| 162 | |
| 163 | if self.dt_rank == "auto": |
| 164 | self.dt_rank = math.ceil(self.dim / 16) |
| 165 | |
| 166 | |
| 167 | class ResidualBlock(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected