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