fix(torch): dataset size is half the database size

jolibrain · Jan 24, 2023 · 9541de1 · 9541de1
1 parent 045aba6
commit 9541de1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/backends/torch/torchdataset.cc b/src/backends/torch/torchdataset.cc
@@ -328,7 +328,7 @@ namespace dd
         if (!_dbCursor)
           _dbCursor = _dbData->NewCursor();
 
-        data_size = _dbData->Count();
+        data_size = _dbData->Count() / 2;
       }
 
     _indices.resize(data_size);