check parameter in XModel

source/train/XLeader.cpp

@@ -138,8 +138,11 @@ void XLeader::SetMode(XLEADER_MODE myMode)
 /* start the workers */
 /* start the workers */
 void XLeader::Start()
 void XLeader::Start()
 {
 {
+    serverModel.CheckParam();
+
     for (int i = 0; i < jworkers.count; i++) {
     for (int i = 0; i < jworkers.count; i++) {
         XWorkerJob * worker = (XWorkerJob*)jworkers.GetItem(i);
         XWorkerJob * worker = (XWorkerJob*)jworkers.GetItem(i);
+        worker->GetModel()->CheckParam();
         worker->Start();
         worker->Start();
     }
     }
 
 

source/train/XModel.cpp

@@ -103,7 +103,7 @@ add a parameter tensor
 */
 */
 void XModel::AddParam(XTensor* param)
 void XModel::AddParam(XTensor* param)
 {
 {
-    //param->SetVarFlag();
+    param->SetVarFlag();
     params.Add(param);
     params.Add(param);
 
 
     PARAM_STATE * newFlags = new PARAM_STATE[params.count];
     PARAM_STATE * newFlags = new PARAM_STATE[params.count];
@@ -115,6 +115,18 @@ void XModel::AddParam(XTensor* param)
     flags = newFlags;
     flags = newFlags;
 }
 }
 
 
+/* check if the parameters are well-defined for training */
+bool XModel::CheckParam()
+{
+    for (int i = 0; i < params.count; i++) {
+        XTensor * param = (XTensor*)params[i];
+        if (!param->isGrad)
+            return false;
+    }
+
+    return true;
+}
+
 /* refresh the model */
 /* refresh the model */
 void XModel::RefreshMe()
 void XModel::RefreshMe()
 {
 {

source/train/XModel.h

@@ -90,7 +90,10 @@ protected:
 
 
 public:
 public:
     /* add a parameter tensor */
     /* add a parameter tensor */
-    void AddParam(XTensor* param);
+    void AddParam(XTensor * param);
+
+    /* check if the parameters are well-defined for training */
+    bool CheckParam();
 
 
     /* refresh the model */
     /* refresh the model */
     void RefreshMe();
     void RefreshMe();