Question

我有一个多类分类任务，我试图使用＆＃39; trainSoftmaxLayer＆＃39;在Matlab中，但它是一个CPU实现版本，并且很慢。因此，我尝试阅读GPU选项的文档，例如＆＃39; trainSoftmaxLayer（＆＃39; useGPU＆＃39;，＆＃39; yes＆＃39;）＆＃39;在传统的神经网络中，但没有任何相关的选择。

Answer 1

最后，通过破解由MATLAB提供的trainSoftmaxLayer.m的源代码来解决这个问题。我们可以编写我们自己的支持GPU的softmax层，如下所示：

function [net] = trainClassifier(x, t, use_gpu, showWindow)
net = network;

% define topology
net.numInputs = 1;
net.numLayers = 1;
net.biasConnect = 1;
net.inputConnect(1, 1) = 1;
net.outputConnect = 1;

% set values for labels
net.name = 'Softmax Classifier with GPU Option';
net.layers{1}.name = 'Softmax Layer';

% define transfer function
net.layers{1}.transferFcn = 'softmax';

% set parameters
net.performFcn = 'crossentropy';
net.trainFcn = 'trainscg';
net.trainParam.epochs = 1000;
net.trainParam.showWindow = showWindow;
net.divideFcn = 'dividetrain';

if use_gpu == 1
    net = train(net, x, full(t), 'useGPU', 'yes');
else
    net = train(net, x, full(t));
end

end

MATLAB中是否有GPU实现多类分类功能？

1 个答案: