从 Keras 中的自定义图层打印

Question

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

Answer 1

试试 `model.summary()' 函数。它会打印有关您的图层尺寸的完整信息。