fix resize bug

vladtimm · Sep 7, 2018 · 1ae3bd5 · 1ae3bd5
1 parent 0b83eb1
commit 1ae3bd5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/framework/core/net/net.cpp b/framework/core/net/net.cpp
@@ -393,9 +393,11 @@ void Net<Ttype, Dtype, Ptype, RunType>::prediction() {
 #define RECORD_INNER
 #if defined(RECORD_INNER) && defined(USE_X86_PLACE)
         record_tensor_to_file(*out,("record_"+executer.name).c_str());
-        if(executer.name=="")
 #endif
         LOG(INFO) <<executer.name <<" d_tensor_out_p :" <<out->data();
+#ifdef USE_CUDA
+    record_tensor_to_file(*out,("record_"+executer.name).c_str());
+#endif
 #ifdef USE_X86_PLACE
 //        for (int i = 0; i < 10; ++i) {
 //            std::cout << out->data()[i]<<" ";