bug fixes

source/train/TTrain.cpp

@@ -39,6 +39,8 @@
 
 namespace nts { // namespace nts(NiuTrans.Tensor)
 
+XTensor * tmpTT = NULL;
+
 /* genreate the training data file */
 void GeneateTTrainData(const char * fileName)
 {
@@ -83,6 +85,8 @@ void TestTrain()
     TTModel model;
     model.Init(config, -1);
 
+    tmpTT = (XTensor*)model.params[0];
+
     XOptimizer optimizer;
     optimizer.Init(config);
 

source/train/TTrain.h

@@ -52,6 +52,8 @@ namespace nts { // namespace nts(NiuTrans.Tensor)
 #define TT_EMBEDDING_SIZE 256
 #define TT_HIDDEN_SIZE 256
 
+extern XTensor * tmpTT;
+
 /* genreate the training data file */
 void GeneateTTrainData(const char * fileName);
 

source/train/XOptimizer.cpp

@@ -87,7 +87,7 @@ void XOptimizer::UpdateParam(XTensor * param, XTensor * grad, int pid)
 {
     /* the delta rule
        \theta_new = \theta_old - \grad * \lrate */
-    Sum(*param, *grad, *param, -lrate);
+    _Sum(param, grad, param, -lrate);
 }
 
 }

source/train/XWorkerBroadcast.cpp

@@ -118,7 +118,7 @@ void XWorkerBroadcast::BroadcastP2P(XTensor * source, XTensor * target)
 {
     CheckNTErrors(source != NULL, "The source tensor should not be NULL!");
     CheckNTErrors(target != NULL, "The target tensor should not be NULL!");
-    CheckNTErrors(IsSameShaped(source, target), "The two tensors should be of the same shape!");
+    CheckNTErrors(IsSameShaped(*source, *target), "The two tensors should be of the same shape!");
 
     CopyValues(*source, *target);
 }

source/train/XWorkerCollect.cpp

@@ -200,7 +200,7 @@ void XWorkerCollect::CollectP2P(XTensor * source, XTensor * target)
 {
     CheckNTErrors(source != NULL, "The source tensor should not be NULL!");
     CheckNTErrors(target != NULL, "The target tensor should not be NULL!");
-    CheckNTErrors(IsSameShaped(source, target), "The two tensors should be of the same shape!");
+    CheckNTErrors(IsSameShaped(*source, *target), "The two tensors should be of the same shape!");
 
     /* target += source */
     if(source != target)