Updated mxnetapi.cpp

* fixed shape for auxiliary outputs
QueensGambit · Aug 26, 2021 · 0f3d60f · 0f3d60f
1 parent 87672f0
commit 0f3d60f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/engine/src/nn/mxnetapi.cpp b/engine/src/nn/mxnetapi.cpp
@@ -157,7 +157,7 @@ void MXNetAPI::init_nn_design()
     set_shape(nnDesign.valueOutputShape, executor->outputs[nnDesign.valueOutputIdx].GetShape());
     nnDesign.hasAuxiliaryOutputs = executor->outputs.size() > 2;
     if (nnDesign.hasAuxiliaryOutputs) {
-        set_shape(nnDesign.valueOutputShape, executor->outputs[nnDesign.auxiliaryOutputIdx].GetShape());
+        set_shape(nnDesign.auxiliaryOutputShape, executor->outputs[nnDesign.auxiliaryOutputIdx].GetShape());
     }
 
     float* inputPlanes = new float[batchSize*StateConstants::NB_VALUES_TOTAL()];