Skip to content

Commit c55cfc2

Browse files
committed
Bug-fix for class label application
1 parent 57862a6 commit c55cfc2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

GPU/GPUTracking/Global/GPUChainTrackingClusterizer.cxx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -967,7 +967,7 @@ int32_t GPUChainTracking::RunTPCClusterizer(bool synchronizeOutput)
967967

968968
auto start1 = std::chrono::high_resolution_clock::now();
969969
GPUTPCNNClusterizer::applyNetworkClass(clusterer, evalDtype);
970-
if (clusterer.model_class.getNumOutputNodes()[0][1] > 1){
970+
if (clusterer.model_class.getNumOutputNodes()[0][1] == 1){
971971
runKernel<GPUTPCNNClusterizer, GPUTPCNNClusterizer::determineClass1Labels>({GetGrid(iSize, lane, GPUReconstruction::krnlDeviceType::CPU), {iSlice}}, evalDtype, 0, batchStart); // Assigning class labels
972972
} else {
973973
runKernel<GPUTPCNNClusterizer, GPUTPCNNClusterizer::determineClass2Labels>({GetGrid(iSize, lane, GPUReconstruction::krnlDeviceType::CPU), {iSlice}}, evalDtype, 0, batchStart); // Assigning class labels

0 commit comments

Comments
 (0)