寻找RESNET的训练脚本在tensorflow cifar10 [关闭]

问题描述 投票:0回答:1

我想找到在tensorflow cifar10 RESNET培训脚本。我已经检查的估计和超薄的代码。但两者都不适合我的项目。

我需要以下条件:

  1. 一个良好的培训脚本,可以达到93%的准确率。 II找到cifar10 RESNET的培训脚本估计是好的。关键是学习率。但估计API是固定的。
  2. 我需要修改模型架构,在模型替换一些欢声笑语。
  3. 控制输入​​数据,这样我可以与其它数据集如SVHN替换cifar10。
  4. 训练结束后,我需要分析每个输入图像的输出Logit模型在训练和测试集。

测试一个新的脚本需要我的电脑上几天。我不想从头开始写剧本。这对我来说是如此昂贵。有人能帮我吗?

tensorflow resnet
1个回答
1
投票

这从tensorpack示例脚本可能会为你工作:https://github.com/tensorpack/tensorpack/blob/master/examples/ResNet/cifar10-resnet.py

  1. 它达到93%+ cifar10精度。
  2. 该架构是在同一个文件(单个文件),这样你可以修改。
  3. 您可以切换到SVHN数据集轻松以下another example或写自己的DataLoader为自己的数据集。
  4. 获取logits需要several lines of code
© www.soinside.com 2019 - 2024. All rights reserved.