faster network test

marius-team · Nov 22, 2023 · eb516eb · eb516eb
1 parent f5ecbb3
commit eb516eb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/cpp/src/pipeline/pipeline_gpu.cpp b/src/cpp/src/pipeline/pipeline_gpu.cpp
@@ -14,7 +14,7 @@
 
 void batchToDevice(Pipeline* pipeline_, shared_ptr<Batch> batch) {
     if (batch->sub_batches_.size() > 0) {
-        #pragma omp parallel for
+//        #pragma omp parallel for
         for (int i = 0; i < batch->sub_batches_.size(); i++) {
             batch->sub_batches_[i]->to(pipeline_->model_->device_models_[i]->device_, pipeline_->dataloader_->compute_streams_[i]);
         }