resnet模型中的层数

Question

我有一个resnet模型的检查点文件。

resnet_v1_101.ckpt

现在，我需要找到这个检查点文件中的层数以及有关这些层的详细信息。

我怎样才能在丹索尔·弗洛做到这一点？

Answer 1

检查点文件没有图形信息。因此，您无法获得有关层的详细信息。但是，如果您通过tf.train.Saver().save(...)保存了您的模型，并且没有指定write_meta_graph=False，那么您应该有一个类似于"resnet_v1_101.ckpt.meta“的文件。这是MetaGraphDef文件。您可以这样导入它：

tf.train.import_meta_graph('./resnet_v1_101.ckpt.meta')

现在，您可以将图形可视化在张表中，也可以使用自己的方式获取有关该图形的信息。