代码之家 › 专栏 › 技术社区 › Freelancer

如何在GPU上强制MATLAB运行CNN

deep-learning matlab

1

Freelancer · 技术社区 · 7 年前

我写了一个程序来训练我的数据(图像与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 列车网络

0 回复 | 直到 7 年前