@@ -27,15 +27,16 @@ GPUTPCNNClusterizerHost::GPUTPCNNClusterizerHost(const GPUSettingsProcessingNNcl
2727 init (settings);
2828}
2929
30- void GPUTPCNNClusterizerHost::loadFromCCDB (std::map<std::string, std::string> settings) {
30+ void GPUTPCNNClusterizerHost::loadFromCCDB (std::map<std::string, std::string> settings)
31+ {
3132 o2::ccdb::CcdbApi ccdbApi;
3233 ccdbApi.init (settings[" nnCCDBURL" ]);
3334
3435 metadata[" inputDType" ] = settings[" inputDType" ];
3536 metadata[" outputDType" ] = settings[" outputDType" ];
36- metadata[" nnCCDBEvalType" ] = settings[" nnCCDBEvalType" ]; // classification_1C, classification_2C, regression_1C, regression_2C
37+ metadata[" nnCCDBEvalType" ] = settings[" nnCCDBEvalType" ]; // classification_1C, classification_2C, regression_1C, regression_2C
3738 metadata[" nnCCDBWithMomentum" ] = settings[" nnCCDBWithMomentum" ]; // 0, 1 -> Only for regression model
38- metadata[" nnCCDBLayerType" ] = settings[" nnCCDBLayerType" ]; // FC, CNN
39+ metadata[" nnCCDBLayerType" ] = settings[" nnCCDBLayerType" ]; // FC, CNN
3940 if (settings[" nnCCDBInteractionRate" ] != " " && std::stoi (settings[" nnCCDBInteractionRate" ]) > 0 ) {
4041 metadata[" nnCCDBInteractionRate" ] = settings[" nnCCDBInteractionRate" ];
4142 }
@@ -58,16 +59,15 @@ void GPUTPCNNClusterizerHost::init(const GPUSettingsProcessingNNclusterizer& set
5859 std::string class_model_path = settings.nnClassificationPath , reg_model_path = settings.nnRegressionPath ;
5960 std::vector<std::string> reg_model_paths;
6061
61- if (settings.nnLoadFromCCDB ) {
62+ if (settings.nnLoadFromCCDB ) {
6263 std::map<std::string, std::string> ccdbSettings = {
6364 {" nnCCDBURL" , settings.nnCCDBURL },
6465 {" nnCCDBPath" , settings.nnCCDBPath },
6566 {" inputDType" , settings.nnInferenceInputDType },
6667 {" outputDType" , settings.nnInferenceOutputDType },
6768 {" nnCCDBWithMomentum" , std::to_string (settings.nnCCDBWithMomentum )},
6869 {" nnCCDBBeamType" , settings.nnCCDBBeamType },
69- {" nnCCDBInteractionRate" , std::to_string (settings.nnCCDBInteractionRate )}
70- };
70+ {" nnCCDBInteractionRate" , std::to_string (settings.nnCCDBInteractionRate )}};
7171
7272 std::string nnFetchFolder = " " ;
7373 std::vector<std::string> fetchMode = o2::utils::Str::tokenize (settings.nnCCDBFetchMode , ' :' );
0 commit comments