updates of XLeader

source/trainer/XLeader.cpp

@@ -39,6 +39,8 @@ namespace nts {
 /* constructor */
 /* constructor */
 XLeader::XLeader()
 XLeader::XLeader()
 {
 {
+    id = -1;
+    dataLoader = NULL;
 }
 }
 
 
 /* de-constructor */
 /* de-constructor */
@@ -58,4 +60,11 @@ int XLeader::GetID()
     return id;
     return id;
 }
 }
 
 
+/* set the data loader */
+void XLeader::SetDataLoader(DataLoaderBase* myDataLoader)
+{
+    CheckNTErrors(myDataLoader != NULL, "The input of XLeader::SetDataLoader should not be NULL!");
+    dataLoader = myDataLoader;
+}
+
 } /* end of the nts (NiuTrans.Tensor) namespace */
 } /* end of the nts (NiuTrans.Tensor) namespace */
\ No newline at end of file

source/trainer/XLeader.h

@@ -35,8 +35,19 @@
 #ifndef __XLEADER_H__
 #ifndef __XLEADER_H__
 #define __XLEADER_H__
 #define __XLEADER_H__
 
 
+#include "XNetTemplate.h"
+
 namespace nts { // namespace nts(NiuTrans.Tensor)
 namespace nts { // namespace nts(NiuTrans.Tensor)
 
 
+/* 
+conmmunication mode of a leader. This offers a way of organizing a hierachy of the work
+1) run as a standalone program
+2) give orders to another leader (probably remote)
+3) recieve orders from anothe leader (probably remote)
+4) give (and recieve) orders to (and from) different leaders
+*/
+enum XLEADER_MODE { XLEADER_STANDALONE, XLEADER_SEND, XLEADER_RECIEVE, XLEADER_SEND_AND_RECIEVE };
+
 /* a leader who manages workers */
 /* a leader who manages workers */
 class XLeader
 class XLeader
 {
 {
@@ -44,6 +55,10 @@ protected:
     /* id of the leader */
     /* id of the leader */
     int id;
     int id;
 
 
+    /* data loader */
+    DataLoaderBase* dataLoader;
+
+
 public:
 public:
     /* constructor */
     /* constructor */
     XLeader();
     XLeader();
@@ -58,7 +73,8 @@ public:
     int GetID();
     int GetID();
 
 
     /* set the data loader */
     /* set the data loader */
-    void SetDataLoader();
+    void SetDataLoader(DataLoaderBase * myDataLoader);
+
 };
 };
 
 
 }
 }