compute timing

marius-team · Nov 21, 2023 · b3e2680 · b3e2680
1 parent ce6ffb0
commit b3e2680
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/cpp/src/data/dataloader.cpp b/src/cpp/src/data/dataloader.cpp
@@ -705,7 +705,7 @@ void DataLoader::loadCPUParameters(shared_ptr<Batch> batch, int id, bool load) {
                             }
 
 //                            batch->sub_batches_[i]->node_features_ = unique_features.narrow(0, count, size);
-                            batch->sub_batches_[i]->node_features_ = graph_storage_->getNodeFeatures(unique_indices.narrow(0, count, size));
+                            batch->sub_batches_[i]->node_features_ = graph_storage_->getNodeFeatures(unique_indices.narrow(0, start, size));
                         }
                     }