从Keras的自定义图层打印

Question

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

Answer 1

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