Matlab:没有足够的GPU内存用于分类
我已经从头开始用我自己的类别(现在只是花类型)培训了Mathworks提供的Resnet50,并且想要测试它。Matlab:没有足够的GPU内存用于分类
虽然在将训练选项中的MiniBatchSize更改为5而不是10后,训练没有问题,但分类不会启动,因为它表示没有足够的内存可用。我正在使用imageDataset中的测试图像,就像在培训期间一样。
有没有办法强制CPU分类或调整一些选项,以便它能够运行? 欢迎任何其他想法!
回答:
我自己实际解决了这个问题:既分类和预测采用名称 - 值对参数,如'MiniBatchSize'或'ExecutionEnvironment'。我能够用这些东西解决我的问题,并修改值。
以上是 Matlab:没有足够的GPU内存用于分类 的全部内容, 来源链接: utcz.com/qa/258652.html