使用我的数据集进行预训练模型的训练DIGITS无法正常工作。

时间:2016-10-20 09:28:13

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

我尝试将预训练模型(VGG 16)用于我的数据集的DIGITS,该数据集有两个类。但是我接受了错误的训练。 (我将最后一层的名称更改为fc8_new) does not connect to previous layers
我已将VGG_ILSVRC_16_layers.caffemodelsynset_words.txt以及$scope.$apply(function() { $scope.loggedIn = true; }); 成功上传到DIGITS,并使用包含两个类的数据集进行测试。  at the end of training

1 个答案:

答案 0 :(得分:1)

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

在我回答here时,您将需要创建一个“一体化”网络,其中包括在单个网络中进行培训,验证和部署所需的所有层。 @Shai也与this helpful answer相关联。

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