updates

source/train/XLeader.cpp

@@ -423,7 +423,19 @@ void XLeader::DestroyParamMap()
 void XLeader::MakeParamMap()
 {
     DestroyParamMap();
-    for(int i = 0; i < jworkers.count; i++){
+    
+    modelNum = jworkers.count;
+    paramMap = new XTensorKeeper*[serverModel.paramNum];
+    gradMap  = new XTensorKeeper*[serverModel.paramNum];
+    
+    for(int i = 0; i < serverModel.paramNum; i++){
+        paramMap[i] = new XTensorKeeper[modelNum];
+        gradMap[i]  = new XTensorKeeper[modelNum];
+        for(int j = 0; j < modelNum; j++){
+            XModel * model = ((XWorkerJob*)jworkers[j])->GetModel();
+            paramMap[i][j].tensor = model->params[j].tensor;
+            gradMap[i][j].tensor  = model->params[j].tensor;
+        }
     }
 }