我已经训练了Mathworks从头开始提供的Resnet50和我自己的类别(现在只是花型)并想测试它。
虽然在trainingOptions中将MiniBatchSize更改为5而不是10之后训练没有问题,但分类将无法启动,因为它表示没有足够的可用内存。我正在使用imageDataset中的测试图像,就像在训练期间一样。
有没有办法强制CPU分类或调整一些选项以便能够运行? 欢迎任何其他想法!
答案 0 :(得分:0)
我实际上自己解决了这个问题:分类和预测采取名称 - 值对参数,例如' MiniBatchSize'或者' ExecutionEnvironment'。我能够使用这些来解决我的问题并修改价值观。