类库
› loss-landscapes
marcellodebernardi/loss-landscapes
这是一个基于PyTorch的Python库,用于在低维子空间中近似神经网络损失函数及相关指标。它简化了损失景观可视化的生成,有助于分析网络损失几何结构。注意:该库处于早期开发阶段,API不稳定且存在潜在Bug,仅限原型设计和研究使用,不支持生产环境。
技术栈
根目录 python
查看全部依赖 (5)
依赖
NumPy
matplotlib
torch
torchvision
tqdm