Skip to content

GitLab

Commit fcc28e95 authored by Ben Graham's avatar Ben Graham Committed by GitHub
Browse files

fix stride

parent 784994a5
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
PyTorch/sparseconvnet/SCN/generic/CPU/SparseToDense.cpp
View file @ fcc28e95
...@@ -36,7 +36,7 @@ extern "C" void scn_DR_(SparseToDense_updateOutput)(THLongTensor *inputSize, ...@@ -36,7 +36,7 @@ extern "C" void scn_DR_(SparseToDense_updateOutput)(THLongTensor *inputSize,
uInt nHot = r.size() / 2; uInt nHot = r.size() / 2;
SparseToDense_ForwardPass<real>(iF, oF, nPlanes, spatialVolume, &r[0], SparseToDense_ForwardPass<real>(iF, oF, nPlanes, spatialVolume, &r[0],
nHot); nHot);
oF += spatialVolume; oF++;
} }
} }
extern "C" void scn_DR_(SparseToDense_updateGradInput)( extern "C" void scn_DR_(SparseToDense_updateGradInput)(
...@@ -58,7 +58,7 @@ extern "C" void scn_DR_(SparseToDense_updateGradInput)( ...@@ -58,7 +58,7 @@ extern "C" void scn_DR_(SparseToDense_updateGradInput)(
uInt nHot = r.size() / 2; uInt nHot = r.size() / 2;
SparseToDense_BackwardPass<real>(diF, doF, nPlanes, spatialVolume, &r[0], SparseToDense_BackwardPass<real>(diF, doF, nPlanes, spatialVolume, &r[0],
nHot); nHot);
doF += spatialVolume; doF++;
} }
} }
#endif #endif
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment