【问题标题】:The training DIGITS for pretrained model with my data-set does not work properly.使用我的数据集进行预训练模型的训练 DIGITS 无法正常工作。
【发布时间】:2017-03-02 05:21:36
【问题描述】:

我尝试将预训练模型 (VGG 16) 用于我的数据集的 DIGITS,该数据集有两个类。但我得到了错误的训练。 (我把最后一层的名字改成 fc8_new)
我将deploy.prototxtVGG_ILSVRC_16_layers.caffemodelsynset_words.txt 成功上传到DIGITS 并用我的数据集进行了测试,该数据集有两个类。

【问题讨论】:

  • 你的问题不是很清楚。您在培训或测试方面遇到问题?
  • 为什么你的第二张图片没有曲线?快照间隔是多少???

标签: machine-learning computer-vision deep-learning caffe nvidia-digits


【解决方案1】:

deploy.prototxt 不会包含任何损失层或准确度层。当您尝试训练没有任何损失层的网络时,我不确定 Caffe 会做什么。

正如我回答 here 一样,您将需要创建一个“一体式”网络,其中包括在单个网络中训练、验证和部署所需的所有层。 @Shai 也链接到 this helpful answer

正如on our GitHub README 解释的那样,我们更愿意在user group 上回答这些问题,而不是在SO 上。

【讨论】:

    猜你喜欢
    • 2017-08-19
    • 2016-10-19
    • 2021-07-09
    • 2020-01-12
    • 2020-05-27
    • 2020-10-26
    • 1970-01-01
    • 2019-01-20
    • 2020-05-06
    相关资源
    最近更新 更多