Tensorflow-Project-Template

收录理由

每次启动一个新的 TensorFlow 模型，往往要重复处理会话、保存点、数据加载和日志这些琐碎事务。这个模板把常见的样板代码整理成一套带抽象基类的目录结构，你只需继承几个类，而不是每次都重写一遍。检查点管理、计数器、摘要日志以及可选的 Comet.ml 上报功能都已备好，要定义自己的模型和训练循环，只需要覆盖两个方法。对于打算在面向对象的 TensorFlow 1.x 代码库上复用模式的人，或者想快速搭建类似 VGG 风格网络的场景，它都提供了一条省事的路径。如果正在学习如何组织一个深度学习项目，示例文件也值得仔细读一读。