From ad9aa4d213f2154e0827c23a942dd8827e9f9c8b Mon Sep 17 00:00:00 2001
From: Marcin Kirsz <mkirsz@ed.ac.uk>
Date: Wed, 16 Oct 2024 10:21:59 +0100
Subject: [PATCH] Removed trained flag in this case as it was misleading
---
trainer.h | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/trainer.h b/trainer.h
index 950aa80..39b55eb 100644
--- a/trainer.h
+++ b/trainer.h
@@ -284,7 +284,7 @@ class MPI_Trainer: public Trainer {
if (rank==0) {
t_type w(dm.T.ptr(), PHI_cols);
model->set_weights(w);
- model->trained=true;
+ // model->trained=true; // still can't train with this model
}
delete[] work2;
MPI_Type_free(&rowvec);
--
GitLab