bug fixes

source/tensor/XQueue.cpp

@@ -215,6 +215,7 @@ void XQueue::DequeueJobs(XList * args)
     int devID = *(int*)args->GetItem(1);
     int devID = *(int*)args->GetItem(1);
 
 
     int devIDBackup = -1;
     int devIDBackup = -1;
+    if(devID >= 0)
         XDevice::SetDevice(devID, devIDBackup);
         XDevice::SetDevice(devID, devIDBackup);
 
 
     while(1){
     while(1){
@@ -236,6 +237,7 @@ void XQueue::DequeueJobs(XList * args)
 
 
     }
     }
 
 
+    if(devID >= 0)
         XDevice::SetDevice(devIDBackup);
         XDevice::SetDevice(devIDBackup);
 }
 }
 
 

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);