文件名称：TensorflowFramework：使用TensorFlow训练神经网络模型的可扩展框架
文件大小：35KB
文件格式：ZIP
更新时间：2024-02-25 23:25:19
machine-learning framework deep-learning neural-network tensorflow
Tensorflow培训框架 该存储库包括一个可扩展框架，用于使用TensorFlow训练神经网络模型。 该框架处理诸如并行数据读取，批处理，改组，保存/加载检查点和日志记录之类的事情。 它还支持在多个GPU上进行训练。 您可以按照数据集/mnist.py中的示例简单地添加自己的数据集，并在model / alexnet.py中编写与捆绑的CNN分类器类似的自己的分类器。 该框架为您完成了大多数样板代码，让您专注于开发实际的神经网络模型。 另外，请确保签出 ，在某种程度上说明了此框架中的代码。 欢迎使用具有新数据集或模型的拉取请求！ 安装依赖 pip install tensorflow
【文件预览】：
TensorflowFramework-master
----model()
--------__init__.py(26B)
--------alexnet.py(2KB)
----LICENSE(1KB)
----dataset()
--------mnist.py(3KB)
--------__init__.py(83B)
--------cifar100.py(3KB)
--------cifar10.py(3KB)
----__init__.py(0B)
----trainer.py(6KB)
----common()
--------io_utils.py(2KB)
--------summary.py(3KB)
--------dataset.py(1KB)
--------__init__.py(0B)
--------ops()
--------model.py(808B)
--------config.py(2KB)
--------hooks.py(8KB)
--------image_utils.py(5KB)
--------optimizer.py(6KB)
--------misc_utils.py(2KB)
----configs()
--------mnist.json(47B)
----.gitignore(35B)
----README.md(1KB)