(m)
| 2 | import time |
| 3 | |
| 4 | def weights_init(m): |
| 5 | classname = m.__class__.__name__ |
| 6 | if classname.find('Conv2d') != -1: |
| 7 | torch.nn.init.xavier_normal_(m.weight.data) |
| 8 | torch.nn.init.constant_(m.bias.data, 0.0) |
| 9 | elif classname.find('Linear') != -1: |
| 10 | torch.nn.init.xavier_normal_(m.weight.data) |
| 11 | torch.nn.init.constant_(m.bias.data, 0.0) |
| 12 | |
| 13 | def weights_init1(m): |
| 14 | classname = m.__class__.__name__ |
nothing calls this directly
no outgoing calls
no test coverage detected