gpt4 book ai didi

machine-learning - 从 Keras 中的自定义层进行打印

转载 作者:行者123 更新时间:2023-11-30 09:09:29 25 4
gpt4 key购买 nike

我目前正在致力于在 Keras 中实现 AlexNet。最近,我遇到了一个关于此的实现 repo这是 AlexNet 架构的一个非常干净的实现。但我觉得这个line上的可用代码存在一些错误。 。如果我以某种方式知道打印变量尺寸的方法,那么我将能够理解并解决该错误。我尝试了 print 函数并写入文件,但它们似乎都不起作用。有谁知道我们可以做什么来从 Keras 库中的此类自定义层打印一些内容?

最佳答案

尝试“model.summary()”函数。它打印有关层尺寸的完整信息。

关于machine-learning - 从 Keras 中的自定义层进行打印,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/43411244/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com