在张量流中寻找cifar10的resnet训练脚本

Question

我想在tensorflow中的cifar10上找到resnet的训练脚本。 我已经检查了estimator和slim中的代码。但是两者都不适合我的项目。

我需要以下内容：

1. 一个好的训练脚本，可以达到93％的准确性。 II发现估计器中在cifar10上使用resnet训练脚本是好的。关键是学习率。但是估算器API是固定的。

2. 我需要修改模型架构，替换模型中的某些操作。

3. 控制输入数据，以便可以将cifar10替换为其他数据集，例如SVHN。

4. 训练后，我需要分析训练和测试集中每个输入图像的输出logit。

在计算机上测试新脚本需要几天的时间。我不想从头开始编写脚本。对我来说太贵了。 有人可以帮我吗？

Answer 1

这个来自tensorpack示例的脚本可能对您有用：https://github.com/tensorpack/tensorpack/blob/master/examples/ResNet/cifar10-resnet.py

1. 达到cifar10 +的93％以上的精度。
2. 体系结构位于同一文件（单个文件）中，因此您可以进行修改。
3. 您可以在another example之后轻松地切换到SVHN数据集，或者为自己的数据集编写自己的数据加载器。
4. 获取登录信息需要several lines of code。