delete resize

deephealthproject · Jan 20, 2021 · 19a50a6 · 19a50a6
1 parent 5ad0c83
commit 19a50a6
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 6 deletions.
diff --git a/examples/nn/4_NLP/3_nlp_machine_translation.cpp b/examples/nn/4_NLP/3_nlp_machine_translation.cpp
@@ -91,9 +91,9 @@ int main(int argc, char **argv) {
           opt, // Optimizer
           {"softmax_cross_entropy"}, // Losses
           {"accuracy"}, // Metrics
-          //CS_GPU({1}) // one GPU
+          CS_GPU({1}) // one GPU
           //CS_GPU({1,1},100) // two GPU with weight sync every 100 batches
-          CS_CPU()
+          //CS_CPU()
     );
 
 

diff --git a/examples/nn/4_NLP/5_nlp_text_generation.cpp b/examples/nn/4_NLP/5_nlp_text_generation.cpp
@@ -66,7 +66,7 @@ int main(int argc, char **argv) {
     download_flickr();
 
     // Settings
-    int epochs = 0;
+    int epochs = 100;
     int batch_size = 24;
 
     int olength=20;
@@ -120,9 +120,9 @@ int main(int argc, char **argv) {
           opt, // Optimizer
           {"softmax_cross_entropy"}, // Losses
           {"accuracy"}, // Metrics
-          //CS_GPU({1}) // one GPU
+          CS_GPU({1}) // one GPU
           //CS_GPU({1,1},100) // two GPU with weight sync every 100 batches
-          CS_CPU()
+          //CS_CPU()
     );
 
     // View model

diff --git a/src/tensor/tensor.cpp b/src/tensor/tensor.cpp
@@ -162,6 +162,7 @@ void Tensor::deleteData(){
         else if (this->isGPU())
         {
             gpu_delete_tensor(this->gpu_device, this->ptr);
+            //cout<<"delete here"<<endl;
         }
 #endif
 #ifdef cFPGA
@@ -585,6 +586,6 @@ void Tensor::resize(int b, float *fptr, void *fptr2, bool delete_data) {
     updateShape(new_shape);
     updateSize();
     updateStrides();
-    if (fptr != nullptr && delete_data) deleteData();  // Potential error on layers such as Reshape (passed pointer)
+    if (!isshared && delete_data) deleteData();  // Potential error on layers such as Reshape (passed pointer)
     updateData(fptr, fptr2);
 }