weight_norm_hook.py 7.7 KB