Skip to content
Snippets Groups Projects
Commit fd1f9ca5 authored by Marcin Kirsz's avatar Marcin Kirsz
Browse files

added throw error if -u flag is used with MPI build

parent 2efd979d
No related branches found
No related tags found
2 merge requests!5MPI version of tadah,!3MPI version of Tadah
Pipeline #43314 failed
......@@ -327,12 +327,18 @@ void TadahCLI::subcommand_train() {
outfile.open ("pot.tadah");
outfile << param_file << std::endl;;
// TODO What about uncertainty when MPI is enabled?
// //if(train->count("--uncertainty")) {
// // t_type weights = model->get_weights();
// // t_type unc = model->get_weights_uncertainty();
// // Output(param_file,false).print_train_unc(weights, unc);
// //}
//
if(train->count("--uncertainty")) {
#ifdef TADAH_BUILD_MPI
throw std::runtime_error("--uncertainty flag is not supported by Tadah! MPI build");
#else
t_type weights = model->get_weights();
t_type unc = model->get_weights_uncertainty();
Output(param_file,false).print_train_unc(weights, unc);
#endif
}
if (is_verbose()) std::cout << timer_tot.to_string() << std::endl;
//
}
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment