Function save_layer_norm

python/save.py:23–27

(layer_norm, path)

        save_tensor(linear.bias, 'bias', path)

def save_layer_norm(layer_norm, path):
    pathlib.Path(path).mkdir(parents=True, exist_ok=True)
    save_tensor(layer_norm.weight, 'weight', path)
    save_tensor(layer_norm.bias, 'bias', path)
    save_scalar(layer_norm.eps, 'eps', path)

def save_group_norm(layer_norm, path):
    pathlib.Path(path).mkdir(parents=True, exist_ok=True)