cameron-chen

Branch

git branch             # list local branches
git branch -r          # list remote branches
git branch -a          # list all branches

git branch -d branch_name       # delete local branch (merged)
git branch -D branch_name       # force delete local branch

cameron-chen

import matplotlib.pyplot as plt
import torch

def viz_weight_norm(
    model: torch.nn.Module, 
    norm_type: float = 2.0,
    lay_names: Optional[Sequence[str]] = None,
    style_use: str = 'seaborn-deep',
):
    """Visualize weight norms of a model.