bug fixes

source/tensor/XQueue.cpp

@@ -215,7 +215,8 @@ void XQueue::DequeueJobs(XList * args)
     int devID = *(int*)args->GetItem(1);
     int devID = *(int*)args->GetItem(1);
 
 
     int devIDBackup = -1;
     int devIDBackup = -1;
-    XDevice::SetDevice(devID, devIDBackup);
+    if(devID >= 0)
+        XDevice::SetDevice(devID, devIDBackup);
 
 
     while(1){
     while(1){
         JobQueueNode * node = (JobQueueNode*)q->Dequeue();
         JobQueueNode * node = (JobQueueNode*)q->Dequeue();
@@ -236,7 +237,8 @@ void XQueue::DequeueJobs(XList * args)
 
 
     }
     }
 
 
-    XDevice::SetDevice(devIDBackup);
+    if(devID >= 0)
+        XDevice::SetDevice(devIDBackup);
 }
 }
 
 
 /* get the break flag */
 /* get the break flag */

source/train/TTrain.cpp

@@ -236,7 +236,7 @@ void TTModel::Init(XConfig &myConfig, int devID)
     int hSize = config.GetInt("hsize", TT_HIDDEN_SIZE);
     int hSize = config.GetInt("hsize", TT_HIDDEN_SIZE);
 
 
     InitTensor2D(&embeddingW, vSize, eSize, X_FLOAT, devID);
     InitTensor2D(&embeddingW, vSize, eSize, X_FLOAT, devID);
-    InitTensor2D(&hiddenW, 3 * eSize, eSize, X_FLOAT, devID);
+    InitTensor2D(&hiddenW, 3 * eSize, hSize, X_FLOAT, devID);
 
 
     embeddingW.SetDataRand(-0.1F, 0.1F);
     embeddingW.SetDataRand(-0.1F, 0.1F);
     hiddenW.SetDataRand(-0.1F, 0.1F);
     hiddenW.SetDataRand(-0.1F, 0.1F);