最新最全手机游戏下载!
首页 游戏资讯 游戏攻略 使用了Wandb,一行Python代码实现了Keras模型的可视化。

使用了Wandb,一行Python代码实现了Keras模型的可视化。

时间: 来源:网络 编辑:2u手游网小编 浏览:55

在训练神经网络的过程中,我们可能希望将网络的性能和中间结构可视化。很多视觉代码冗长复杂,让我们望而却步。有没有一行代码可以解决可视化的所有问题?

通过wandb,只需要一行Python代码就可以可视化Keras网络性能指标和结构。(注:Keras使得神经网络的构建简单明了,很受欢迎)

关注,转发,私信边肖“01”免费领取Python学习资料~!

怎么下载这么有用的包?

只需运行“pip install wandb”即可轻松安装wandb,然后就可以运行所有的Keras实例了。

实测

为了测试wandb的可视化效果,我修改了Keras examples目录中的一些脚本来测试这个功能。只需将下面一行代码添加到培训脚本的顶部。

从wandb导入magic keras示例链接:

https://github.com/keras-team/keras

简单CNN

从入门的mnist_cnn.py开始,我加了一行“来自wandb进口魔术”——。也可以查一下mnist_cn.py,是从Keras examples中fork出来的,只改了一行。

Mnist_cn.py链接:

https://github.com/lukas/keras/blob/master/examples/mnist _ CNN . py

当模型运行时,wandb在后台启动一个进程,并将相关指标保存到wandb。您可以访问网页链接来查看输出。结果如下:

链接:

https://app.wandb.ai/l2k2/keras-examples/runs/ovptynun/model

模型的标记数据、损耗曲线和精度曲线如下:

Cifar数据集上的ResNet

接下来,我派生了cifar10_ResNet.py并类似地更改了一行。你可以在wandb.com看到resnet的可视化。

Cifar10_resnet.py链接:

https://github.com/lukas/keras/blob/master/examples/cifar 10 _ resnet . py

可视web链接:

https://app.wandb.ai/l2k2/keras-examples/runs/ieqy2e9h/model

在系统界面上,你可以看到这个模型比mnist例子用了更多的GPU。

系统接口链接:

https://app.wandb.ai/l2k2/keras-examples/runs/ieqy2e9h/system

暹罗网络

接下来我试了一下暹罗网的例子。在这个例子中,我想看看张量流图。幸运的是,调用可视化工具TensorBoard也只需要一行代码。你可以点击下面的链接查看。

可视化后链接:

https://app.wandb.ai/l2k2/keras-examples/runs/fsc63n6a工作空间=用户-l2k2

其他功能

每个模型的测试只需要不到一分钟,并且只增加了一点点计算开销,而且这个方法适用于你使用的任何一个Keras模型。如果想跟踪更多的东西,可以用下面两行代替:

导入wandbwondb . init(magic=true)那么你就可以使用自定义的wandb.log()函数来保存你想要的任何东西。可以去wandb网站了解更多信息。

Wandb网站链接:

https://docs.wandb.com/docs/started.html

相关攻略

猜你喜欢