Function init

methods/SpatialAttGRU.py:12–18

(module, weight_init, bias_init, gain=1)

def init(module, weight_init, bias_init, gain=1):
    weight_init(module.weight.data, gain=gain)
    if module.bias is not None:
        bias_init(module.bias.data)
    else:
        print('None bias')
    return module


init_ = lambda m: init(m,