Error using trainNetwork. Output size does not match the response s... (2024)

Why are you flattening the data with ,

Apologies - that was a carry over from another example I was playing around with. Removing that sees a new error, where there is a difference between the output and response size.

pressuredata = rand(100, 100, 10);

velocitydata = rand(100, 100, 10);

%% Define and prepare the input and output data

inputData = pressuredata; % Set input = pressure data

outputData = velocitydata; % Set output = velocity field

%% Determine input size

inputSize = size(inputData); % 100 x 100 x 10

outputSize = size(outputData); % 100 x 100 x 10

%% Set layer variables

layers = [

imageInputLayer([100 100 10])

fullyConnectedLayer(100) % Adjust the number of neurons in the fully connected layer as needed

reluLayer % Activation function

fullyConnectedLayer(numel(outputData))

reshapeLayer("reshape layer", outputSize)

regressionLayer % Regression layer for continuous output

];

%% analyze network

analyzeNetwork(layers);

%% Define training options

options = trainingOptions('adam', ...

'MaxEpochs', 50, ...

'MiniBatchSize', 32, ...

'Verbose', true);

%% Train network

net = trainNetwork(inputData, outputData, layers, options);

classdef reshapeLayer < nnet.layer.Layer

properties

outputSize;

end

properties (Learnable)

end

methods

function layer = reshapeLayer(name, outputSize)

layer.Name = name;

layer.outputSize = outputSize;

end

function [Z] = predict(layer, X)

Z = reshape(X,layer.outputSize(1), layer.outputSize(2), layer.outputSize(3),[]);

end

end

end