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

Use Trainer for non mpi version

parent 55e7e718
No related branches found
No related tags found
2 merge requests!5MPI version of tadah,!3MPI version of Tadah
Pipeline #43367 failed
...@@ -310,6 +310,7 @@ void TadahCLI::subcommand_train() { ...@@ -310,6 +310,7 @@ void TadahCLI::subcommand_train() {
blacs_gridexit_(&tr.context2); blacs_gridexit_(&tr.context2);
#else // NON MPI VERSION #else // NON MPI VERSION
Trainer tr(config);
if (is_verbose()) std::cout << "Loading structures..." << std::flush; if (is_verbose()) std::cout << "Loading structures..." << std::flush;
StructureDB stdb(tr.config); StructureDB stdb(tr.config);
if (is_verbose()) std::cout << "Done!" << std::endl; if (is_verbose()) std::cout << "Done!" << std::endl;
...@@ -322,7 +323,7 @@ void TadahCLI::subcommand_train() { ...@@ -322,7 +323,7 @@ void TadahCLI::subcommand_train() {
if (is_verbose()) std::cout << "Training start..." << std::flush; if (is_verbose()) std::cout << "Training start..." << std::flush;
// here we want to train with the locally allocated phi // here we want to train with the locally allocated phi
tr.model->train(stdb,dc); tr.model->train(stdb,tr.dc);
if (is_verbose()) std::cout << "Done!" << std::endl; if (is_verbose()) std::cout << "Done!" << std::endl;
...@@ -337,8 +338,8 @@ void TadahCLI::subcommand_train() { ...@@ -337,8 +338,8 @@ void TadahCLI::subcommand_train() {
#ifndef TADAH_BUILD_MPI #ifndef TADAH_BUILD_MPI
if(train->count("--uncertainty")) { if(train->count("--uncertainty")) {
t_type weights = model->get_weights(); t_type weights = tr.model->get_weights();
t_type unc = model->get_weights_uncertainty(); t_type unc = tr.model->get_weights_uncertainty();
Output(param_file,false).print_train_unc(weights, unc); Output(param_file,false).print_train_unc(weights, unc);
} }
#endif #endif
......
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