Merge pull request #47 from sarthakpati/master

cli fix
FeTS-AI · Apr 21, 2021 · cbf9f35 · cbf9f35
2 parents f8e0259 + 5681a95
commit cbf9f35
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/applications/FeTS_CLI.cxx b/src/applications/FeTS_CLI.cxx
@@ -510,7 +510,7 @@ int main(int argc, char** argv)
       fout.close();
     }
 
-    std::string args = " -d " + dataDir + " -ld " + loggingDir + " -col " + colName + device_arg + " -bsuf " + validation_to_send,
+    std::string args = " -d " + dataDir + " -ld " + loggingDir + " -col " + colName + device_arg,
       hardcodedModelName;
 
     if (!patchValidation)
@@ -536,7 +536,7 @@ int main(int argc, char** argv)
     std::string fullCommandToRun = hardcodedPythonPath + " " + fetsApplicationPath;
     fullCommandToRun += "/OpenFederatedLearning/bin/run_collaborator_from_flplan.py";
 
-    auto temp_args = args + " -p " + hardcodedPlanName + ".yaml";
+    auto temp_args = args + " -p " + hardcodedPlanName + ".yaml" + " -bsuf " + validation_to_send;
 
     std::cout << "Starting training...\n";