(m)
| 70 | |
| 71 | # ##################################################################### |
| 72 | def weights_init_kaiming(m): |
| 73 | classname = m.__class__.__name__ |
| 74 | # print(classname) |
| 75 | if classname.find('Conv') != -1: |
| 76 | init.kaiming_normal_(m.weight.data, a=0, mode='fan_in') |
| 77 | elif classname.find('Linear') != -1: |
| 78 | init.kaiming_normal_(m.weight.data, a=0, mode='fan_out') |
| 79 | init.zeros_(m.bias.data) |
| 80 | elif classname.find('BatchNorm1d') != -1: |
| 81 | init.normal_(m.weight.data, 1.0, 0.01) |
| 82 | init.zeros_(m.bias.data) |
| 83 | |
| 84 | def weights_init_classifier(m): |
| 85 | classname = m.__class__.__name__ |
nothing calls this directly
no outgoing calls
no test coverage detected