From 8697b0ef912ff596af204f58701601132bca5a60 Mon Sep 17 00:00:00 2001 From: Julio Jerez Date: Fri, 17 Nov 2023 13:45:27 -0800 Subject: [PATCH] using droput for cifar 10 classifier (wip) --- .../ndSandbox/toolbox/ndCifar10ImageClassification.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/newton-4.00/applications/ndSandbox/toolbox/ndCifar10ImageClassification.cpp b/newton-4.00/applications/ndSandbox/toolbox/ndCifar10ImageClassification.cpp index 572177c211..d22eea30b5 100644 --- a/newton-4.00/applications/ndSandbox/toolbox/ndCifar10ImageClassification.cpp +++ b/newton-4.00/applications/ndSandbox/toolbox/ndCifar10ImageClassification.cpp @@ -419,10 +419,10 @@ static void Cifar10TrainingSet() //layers.PushBack(new ACTIVATION_TYPE(conv->GetOutputSize())); ndInt32 neuronsPerLayers = 128; - layers.PushBack(new ndBrainLayerLinear(layers[layers.GetCount() - 1]->GetOutputSize(), neuronsPerLayers)); + layers.PushBack(new ndBrainLayerLinearWithDropOut(layers[layers.GetCount() - 1]->GetOutputSize(), neuronsPerLayers)); layers.PushBack(new ACTIVATION_TYPE(layers[layers.GetCount() - 1]->GetOutputSize())); - layers.PushBack(new ndBrainLayerLinear(layers[layers.GetCount() - 1]->GetOutputSize(), neuronsPerLayers)); + layers.PushBack(new ndBrainLayerLinearWithDropOut(layers[layers.GetCount() - 1]->GetOutputSize(), neuronsPerLayers)); layers.PushBack(new ACTIVATION_TYPE(layers[layers.GetCount() - 1]->GetOutputSize())); layers.PushBack(new ndBrainLayerLinear(layers[layers.GetCount() - 1]->GetOutputSize(), trainingLabels->GetColumns()));