Skip to content
Snippets Groups Projects
Commit d2836bb0 authored by Cyril Moineau's avatar Cyril Moineau Committed by Nathan Thoumine
Browse files

Fix BatchStream.hpp initialization

parent 56b98f59
No related branches found
No related tags found
No related merge requests found
......@@ -16,7 +16,7 @@ public:
, _prefix(prefix)
{
_dims.nbDims = 4; //The number of dimensions.
_dims.d[0] = 1; //Batch Size
_dims.d[0] = batchSize; //Batch Size
_dims.d[1] = dimC; //Channels
_dims.d[2] = dimY; //Height
_dims.d[3] = dimX; //Width
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment