Fix invocation.

pytorch · Sep 6, 2021 · 526faab · 526faab
1 parent f486437
commit 526faab
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/examples/cpp/hello_world/main.cpp b/examples/cpp/hello_world/main.cpp
@@ -18,19 +18,18 @@ int main()
     std::cout << "Other error: " << e.what() << "\n";
     return -1;
   }
-  model->eval();
 
   // Create a random input tensor and run it through the model.
   auto in = torch::rand({1, 3, 10, 10});
-  auto out = model->forward(in);
+  auto out = model.forward(in);
 
   std::cout << out.sizes();
 
   if (torch::cuda::is_available()) {
     // Move model and inputs to GPU
-    model->to(torch::kCUDA);
+    model.to(torch::kCUDA);
     auto gpu_in = in.to(torch::kCUDA);
-    auto gpu_out = model->forward(gpu_in);
+    auto gpu_out = model.forward(gpu_in);
 
     std::cout << gpu_out.sizes();
   }