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

mem manager

parent d25ec8ba
No related branches found
No related tags found
No related merge requests found
Pipeline #51786 passed
......@@ -8,6 +8,7 @@
#include <tadah/mlip/nn_finder.h>
#include <tadah/core/config.h>
#include <tadah/models/dc_selector.h>
#include <tadah/models/memory/IModelsWorkspaceManager.h>
#include <iostream>
......@@ -27,6 +28,22 @@ class Trainer {
if(fb)
delete fb;
}
Trainer (Config &c, tadah::models::memory::IModelsWorkspaceManager& workspaceManager):
config(c),
DCS(config),
dc(config,*DCS.d2b,*DCS.d3b,*DCS.dmb,
*DCS.c2b,*DCS.c3b,*DCS.cmb),
nnf(config),
fb(CONFIG::factory<DM_Function_Base,Config&>(
config.get<std::string>("MODEL",1),config)),
model(CONFIG::factory<M_Tadah_Base,DM_Function_Base&,Config&>
(config.get<std::string>("MODEL",0),*fb,config,workspaceManager)),
//(config.get<std::string>("MODEL",0),*fb,config)),
dm(*fb, config)
{
config.postprocess();
config.check_for_training();
}
Trainer (Config &c):
config(c),
DCS(config),
......
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