Reset parameters.
(self)
| 434 | self.reset_parameters() |
| 435 | |
| 436 | def reset_parameters(self): |
| 437 | """Reset parameters.""" |
| 438 | nn.Conv2d.reset_parameters(self) |
| 439 | if hasattr(self, "lora_A"): |
| 440 | # initialize A the same way as the default for nn.Linear and B to zero |
| 441 | nn.init.kaiming_uniform_(self.lora_A, a=math.sqrt(5)) |
| 442 | nn.init.zeros_(self.lora_B) |
| 443 | |
| 444 | def train(self, mode: bool = True): |
| 445 | """Train. |
no test coverage detected