(self)
| 85 | |
| 86 | # 定义权值初始化 |
| 87 | def initialize_weights(self): |
| 88 | for m in self.modules(): |
| 89 | if isinstance(m, nn.Conv2d): |
| 90 | torch.nn.init.xavier_normal_(m.weight.data) |
| 91 | if m.bias is not None: |
| 92 | m.bias.data.zero_() |
| 93 | elif isinstance(m, nn.BatchNorm2d): |
| 94 | m.weight.data.fill_(1) |
| 95 | m.bias.data.zero_() |
| 96 | elif isinstance(m, nn.Linear): |
| 97 | torch.nn.init.normal_(m.weight.data, 0, 0.01) |
| 98 | m.bias.data.zero_() |
| 99 | |
| 100 | |
| 101 | net = Net() # 创建一个网络 |
nothing calls this directly
no outgoing calls
no test coverage detected