类库 › loss-landscape
tomgoldstein

tomgoldstein/loss-landscape

该仓库提供基于PyTorch的代码,用于可视化神经网络的损失景观。给定网络架构和预训练参数,工具计算并可视化最优参数附近随机方向上的损失曲面。支持多GPU和多节点并行计算,结果存储为HDF5文件,适用于深度学习模型分析研究。

3,198 440 3,198 30
在 GitHub 上查看

评论

Home - Wiki
Copyright © 2011-2026 iteam. Current version is 2.155.2. UTC+08:00, 2026-08-17 07:34
浙ICP备14020137号-1 $Map of visitor$