0

我已经用我自己的类别（现在只是花卉类型）从头开始训练了 Mathworks 提供的 Resnet50，并想对其进行测试。

尽管在 trainingOptions 中将 MiniBatchSize 更改为 5 而不是 10 后，训练没有问题，但分类不会开始，因为它说没有足够的可用内存。我在 imageDataset 中使用测试图像，就像在训练期间一样。

有什么方法可以强制 CPU 分类或调整某些选项使其能够运行？也欢迎任何其他想法！

4

1 回答 1

0

实际上我自己解决了这个问题：分类和预测都采用名称-值对参数，例如“MiniBatchSize”或“ExecutionEnvironment”。我能够使用这些解决我的问题并修改这些值。

于 2017-12-03T09:28:31.590 回答