(m)
| 79 | |
| 80 | |
| 81 | def weights_init_normal(m): |
| 82 | classname = m.__class__.__name__ |
| 83 | if classname.find('Conv') != -1: |
| 84 | torch.nn.init.normal(m.weight,0.0,0.02) |
| 85 | elif classname.find('BatchNorm2d') != -1: |
| 86 | torch.nn.init.normal(m.weight, 0.0, 0.02) |
| 87 | torch.nn.init.constant(m.bias,0.0) |
| 88 | |
| 89 | if __name__ == '__main__': |
| 90 | gen = Generator() |
nothing calls this directly
no outgoing calls
no test coverage detected