Function grayscale_weight_grid_vis

lib/vis.py:26–28

(w, (nh, nw), save_path=None)

    return img

def grayscale_weight_grid_vis(w, (nh, nw), save_path=None):
    w = (w+w.min())/(w.max()-w.min())
    return grayscale_grid_vis(w, (nh, nw), save_path=save_path)