TensorflowFramework:使用TensorFlow训练神经网络模型的可扩展框架 源码

Tensorflow培训框架 该存储库包括一个可扩展框架,用于使用TensorFlow训练神经网络模型。 该框架处理诸如并行数据读取,批处理,改组,保存/加载检查点和日志记录之类的事情。 它还支持在多个GPU上进行训练。 您可以按照数据集/mnist.py中的示例简单地添加自己的数据集,并在model / alexnet.py中编写与捆绑的CNN分类器类似的自己的分类器。 该框架为您完成了大多数样板代码,让您专注于开发实际的神经网络模型。 另外,请确保签出 ,在某种程度上说明了此框架中的代码。 欢迎使用具有新数据集或模型的拉取请求! 安装依赖 pip install tensorflow