small improvements
This commit is contained in:
parent
284a313751
commit
4e238675c8
@ -55,27 +55,18 @@ static void prepareNeuralNetworkFile(const char *path, const NeuralNetwork nn)
|
|||||||
|
|
||||||
// input und output dimension schreiben
|
// input und output dimension schreiben
|
||||||
int inputDim = nn.layers[0].weights.cols;
|
int inputDim = nn.layers[0].weights.cols;
|
||||||
int outputDim = nn.layers[0].weights.rows;
|
|
||||||
fwrite(&inputDim, sizeof(int), 1, file);
|
fwrite(&inputDim, sizeof(int), 1, file);
|
||||||
fwrite(&outputDim, sizeof(int), 1, file);
|
|
||||||
|
|
||||||
// erstes Layer schreiben
|
|
||||||
int weightCount = nn.layers[0].weights.rows * nn.layers[0].weights.cols;
|
|
||||||
fwrite(nn.layers[0].weights.buffer, sizeof(MatrixType), weightCount, file);
|
|
||||||
|
|
||||||
int biasesCount = nn.layers[0].biases.rows * nn.layers[0].biases.cols;
|
|
||||||
fwrite(nn.layers[0].biases.buffer, sizeof(MatrixType), biasesCount, file);
|
|
||||||
|
|
||||||
// für weiter Layer nur outputDimension schreiben
|
// für weiter Layer nur outputDimension schreiben
|
||||||
for (unsigned int i = 1; i < nn.numberOfLayers; i++)
|
for (unsigned int i = 0; i < nn.numberOfLayers; i++)
|
||||||
{
|
{
|
||||||
outputDim = nn.layers[i].weights.rows;
|
int outputDim = nn.layers[i].weights.rows;
|
||||||
fwrite(&outputDim, sizeof(int), 1, file);
|
fwrite(&outputDim, sizeof(int), 1, file);
|
||||||
|
|
||||||
weightCount = nn.layers[i].weights.rows * nn.layers[i].weights.cols;
|
int weightCount = nn.layers[i].weights.rows * nn.layers[i].weights.cols;
|
||||||
fwrite(nn.layers[i].weights.buffer, sizeof(MatrixType), weightCount, file);
|
fwrite(nn.layers[i].weights.buffer, sizeof(MatrixType), weightCount, file);
|
||||||
|
|
||||||
biasesCount = nn.layers[i].biases.rows * nn.layers[i].biases.cols;
|
int biasesCount = nn.layers[i].biases.rows * nn.layers[i].biases.cols;
|
||||||
fwrite(nn.layers[i].biases.buffer, sizeof(MatrixType), biasesCount, file);
|
fwrite(nn.layers[i].biases.buffer, sizeof(MatrixType), biasesCount, file);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user