Function save_group_norm

python/save.py:29–37

(layer_norm, path)

    save_scalar(layer_norm.eps, 'eps', path)

def save_group_norm(layer_norm, path):
    pathlib.Path(path).mkdir(parents=True, exist_ok=True)
    if layer_norm.weight is not None:
        save_tensor(layer_norm.weight, 'weight', path)
    if layer_norm.bias is not None:
        save_tensor(layer_norm.bias, 'bias', path)
    save_scalar(layer_norm.eps, 'eps', path)
    save_scalar(layer_norm.num_groups, 'n_group', path)
    save_scalar(layer_norm.num_channels, 'n_channel', path)

def to_tuple_tensor(val):
    if isinstance(val, tuple):