tmp code

source/sample/transformer/T2TTrainer.cpp

@@ -88,8 +88,10 @@ void T2TTrainer::Train(const char * fn, T2TModel * model)
     float lr = 0;
 
     int devID = model->devID;
-    XMem * mem = model->mem;
+    XMem * mem = new XMem(devID, UNI_FREE, MILLION * 256, 1024, MILLION * 64);
+
     mem->SetPin();
+    model->mem->SetPin();
 
     XNet net;
     
@@ -102,6 +104,7 @@ void T2TTrainer::Train(const char * fn, T2TModel * model)
         
         wordCount = 0;
 
+        model->mem->BackToPin();
         mem->BackToPin();
         
         /* batch of input sequences */
@@ -150,6 +153,7 @@ void T2TTrainer::Train(const char * fn, T2TModel * model)
                         lr, elapsed, step, epoch + 1, wordCountTotal, exp(loss / wordCount));
             }
 
+            model->mem->BackToPin();
             mem->BackToPin();
         }