예제 #1
0
def reinit_conv2d_parameters(conv2dlayer: nn.Conv2d):
    w, b = conv2dlayer.parameters()
    glorot_reinitialize_kernel_(w)
    b.data[:] = 0