动画化神经网络的优化轨迹 loss-landscape-anim允许您在神经网络的损耗格局的2D切片中创建动画优化路径。 它基于 ,如果要添加自己的模型,请遵循其建议的样式。 请查看我的文章以获取更多示例和一些直观说明。 0.安装 从PyPI: pip install loss-landscape-anim 从源头上讲,您需要。 克隆此存储库后,请运行以下命令以安装依赖项。 poetry install 1.基本范例 使用提供的和默认的多层感知器MLP模型,您可以直接调用loss_landscape_anim来获得示例动画GIF,如下所示: # Use default MLP model and sample spirals dataset loss_landscape_anim ( n_epochs = 300 ) 注意:如果您在笔记本电脑上使用它,请不要忘记在顶部包括以
2022-03-08 16:47:18 73.14MB pytorch pca neural-nets pytorch-lightning
1