我写了一个程序来训练我的数据(图像与trainNetwork)。我用带GPU的电脑 内存64GB用于训练。我在这里写了一个简短的代码:
opts=trainingOptions('sgdm','InitialLearnRate',1e-3,'MaxEpochs',200,'MiniBatchSize',64,'Plots','training-progress', 'ExecutionEnvironment', 'auto'); net=trainNetwork(trainingData,layers,opts);
如Mathworks网站所述 trainNetwork 将在GPU可用时选择它。尽管我试图改变 ExecutionEnvironment 列车网络
trainNetwork
ExecutionEnvironment
列车网络