max/poet
1
0
Fork 0
mirror of https://git.gfz-potsdam.de/naaice/poet.git synced 2025-12-13 03:18:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

update weight path

Browse Source
This commit is contained in:
Hannes Signer 2025-12-11 00:46:52 +01:00
parent 672e1801cd
commit 950793b73e
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/poet.cpp
View File

@ -316,11 +316,9 @@ static Rcpp::List RunMasterLoop(RInsidePOET &R, const RuntimeParameters &params,
if (params.ai) {
if (params.function_code == 1) {
ai_ctx = std::make_unique<AIContext>(
"./bench/barite/barite_trained.weights.h5");
ai_ctx = std::make_unique<AIContext>("./barite_trained.weights.h5");
} else if (params.function_code == 2) {
ai_ctx = std::make_unique<AIContext>(
"./bench/dolo/dolomite_trained.weights.h5");
ai_ctx = std::make_unique<AIContext>("./dolomite_trained.weights.h5");
}
R.parseEval(