using droput for cifar 10 classifier (wip)

MADEAPPS · Nov 17, 2023 · 8697b0e · 8697b0e
1 parent c387d74
commit 8697b0e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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()));