Matlab:没有足够的GPU内存进行分类

时间:2017-12-02 22:40:24

标签: matlab neural-network classification resnet

我已经训练了Mathworks从头开始提供的Resnet50和我自己的类别(现在只是花型)并想测试它。

虽然在trainingOptions中将MiniBatchSize更改为5而不是10之后训练没有问题,但分类将无法启动,因为它表示没有足够的可用内存。我正在使用imageDataset中的测试图像,就像在训练期间一样。

有没有办法强制CPU分类或调整一些选项以便能够运行? 欢迎任何其他想法!

1 个答案:

答案 0 :(得分:0)

我实际上自己解决了这个问题:分类和预测采取名称 - 值对参数,例如' MiniBatchSize'或者' ExecutionEnvironment'。我能够使用这些来解决我的问题并修改价值观。