(module)
| 193 | |
| 194 | def initialize_weights(self): |
| 195 | def _basic_init(module): |
| 196 | if isinstance(module, nn.Linear): |
| 197 | torch.nn.init.xavier_uniform_(module.weight) |
| 198 | if module.bias is not None: |
| 199 | nn.init.constant_(module.bias, 0) |
| 200 | self.apply(_basic_init) |
| 201 | |
| 202 | # Initialize timestep embedding MLP |
nothing calls this directly
no outgoing calls
no test coverage detected