(self, conv)
| 176 | nn.init.zeros_(conv.bias.data) |
| 177 | |
| 178 | def init_weight2(self, conv): |
| 179 | conv_weight = conv.weight.data |
| 180 | nn.init.zeros_(conv_weight) |
| 181 | c1, c2, t, h, w = conv_weight.size() |
| 182 | init_matrix = torch.eye(c1 // 2, c2) |
| 183 | #init_matrix = repeat(init_matrix, 'o ... -> (o 2) ...').permute(1,0,2).contiguous().reshape(c1,c2) |
| 184 | conv_weight[:c1 // 2, :, -1, 0, 0] = init_matrix |
| 185 | conv_weight[c1 // 2:, :, -1, 0, 0] = init_matrix |
| 186 | conv.weight.data.copy_(conv_weight) |
| 187 | nn.init.zeros_(conv.bias.data) |
| 188 | |
| 189 | |
| 190 | class ResidualBlock(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected