remove gpu input when call on cpu

deepmodeling · Mar 28, 2024 · 9b151d5 · 9b151d5
1 parent 811ff3a
commit 9b151d5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/source/api_cc/src/DeepPotPT.cc b/source/api_cc/src/DeepPotPT.cc
@@ -47,7 +47,7 @@ void DeepPotPT::init(const std::string& model,
   gpu_enabled = torch::cuda::is_available();
   if (!gpu_enabled) {
     device = torch::Device(torch::kCPU);
-    std::cout << "load model from: " << model << " to cpu " << gpu_id
+    std::cout << "load model from: " << model << " to cpu " 
               << std::endl;
   } else {
     std::cout << "load model from: " << model << " to gpu " << gpu_id