forgot gradient change

2022-06-28 20:03:40 +05:30
parent 4959e7c5ed
commit 9e7bddf8ce
1 changed files with 2 additions and 2 deletions
@@ -489,12 +489,12 @@ void ConvolutionType<
        else if (gradientTemp.n_rows > output.n_rows ||
                 gradientTemp.n_cols > output.n_cols)
        {
-          gradientTemp.slice(outMap).submat(0, 0, output.n_rows - 1,
+          gradientTemp.slice((outMap * inMaps) + inMap).submat(0, 0, output.n_rows - 1,
              output.n_cols - 1) += output;
        }
        else
        {
-          gradientTemp.slice(outMap) += output;
+          gradientTemp.slice((outMap * inMaps) + inMap) += output;
        }
      }