Matlab：没有足够的GPU内存用于分类

Question

我已经从头开始用我自己的类别（现在只是花类型）培训了Mathworks提供的Resnet50，并且想要测试它。

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

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

Answer 1

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