loss_landscape package

Submodules

loss_landscape.landscape_utils module

loss_landscape.landscape_utils.rand_basis(ws: Dict, device: str | None = device(type='cpu'))

loss_landscape.landscape_utils.normalize_filter(bs: Dict, ws: Dict)

loss_landscape.landscape_utils.ignore_bn(ws: Dict)

loss_landscape.landscape_utils.create_bases(model: Module, device: str | None = device(type='cpu'), has_module: bool | None = False)

loss_landscape.landscape_utils.generate_plots(xx, yy, zz, model_name, results_loc)

loss_landscape.landscape_utils.plot_save_graphs(save_dir: str, model_name: str, grid_a: ndarray, grid_b: ndarray, loss_surface: ndarray, resolution: int)

Module contents