Matlab：没有足够的 GPU 内存进行分类
Matlab: Not enough GPU memory for classification
我已经用我自己的类别（现在只有花卉类型）从头开始训练 Mathworks 提供的 Resnet50，并想对其进行测试。
虽然我在 trainingOptions 中将 MiniBatchSize 更改为 5 而不是 10 后训练没有问题，但分类不会开始，因为它说没有足够的可用内存。我在 imageDataset 中使用测试图像，就像在训练期间一样。
有没有办法强制CPU分类或调整一些选项使其能够运行？
也欢迎任何其他想法！
我实际上自己解决了这个问题：分类和预测都采用名称-值对参数，例如 'MiniBatchSize' 或 'ExecutionEnvironment'。我能够使用这些并修改值来解决我的问题。
我已经用我自己的类别（现在只有花卉类型）从头开始训练 Mathworks 提供的 Resnet50，并想对其进行测试。
虽然我在 trainingOptions 中将 MiniBatchSize 更改为 5 而不是 10 后训练没有问题，但分类不会开始，因为它说没有足够的可用内存。我在 imageDataset 中使用测试图像，就像在训练期间一样。
有没有办法强制CPU分类或调整一些选项使其能够运行？ 也欢迎任何其他想法！
我实际上自己解决了这个问题：分类和预测都采用名称-值对参数，例如 'MiniBatchSize' 或 'ExecutionEnvironment'。我能够使用这些并修改值来解决我的问题。