我一直在寻找在张量流上写的AlexNet模型,而我发现的所有代码都是使用了一些预先训练过的权重。
您是否知道在执行模型期间是否存在构建权重的代码? 感谢。
答案 0 :(得分:4)
你可以在这里找到一篇好文章:Finetuning AlexNet with TensorFlow 它包含github代码的地址
答案 1 :(得分:2)
您可以在TensorFlow存储库的路径tensorflow/contrib/slim/python/slim/nets/alexnet.py
中找到TensorFlow中AlexNet模型的定义(以前是TF-Slim的示例,现在只是tf.contrib.layers
)。< / p>
答案 2 :(得分:0)