Fixing hello-world and removing tests

examples/cpp/hello_world/main.cpp

@@ -1,11 +1,23 @@
 #include <iostream>
+#include <torch/script.h>
 #include <torch/torch.h>
 #include <torchvision/vision.h>
-#include <torchvision/models/resnet.h>
 
 int main()
 {
-  auto model = vision::models::ResNet18();
+  torch::jit::script::Module model;
+  try {
+    std::cout << "Loading model\n";
+    // Deserialize the ScriptModule from a file using torch::jit::load().
+    model = torch::jit::load("resnet18.pt");
+    std::cout << "Model loaded\n";
+  } catch (const torch::Error& e) {
+    std::cout << "error loading the model\n";
+    return -1;
+  } catch (const std::exception& e) {
+    std::cout << "Other error: " << e.what() << "\n";
+    return -1;
+  }
   model->eval();
 
   // Create a random input tensor and run it through the model.

examples/cpp/hello_world/trace_model.py

@@ -0,0 +1,13 @@
+import os.path as osp
+
+import torch
+import torchvision
+
+HERE = osp.dirname(osp.abspath(__file__))
+ASSETS = osp.dirname(osp.dirname(HERE))
+
+model = torchvision.models.resnet18(pretrained=False)
+model.eval()
+
+traced_model = torch.jit.script(model)
+traced_model.save("resnet18.pt")

packaging/build_cmake.sh

@@ -98,8 +98,12 @@ fi
 # Compile and run the CPP example
 popd
 cd examples/cpp/hello_world
-
 mkdir build
+
+# Trace model
+python trace_model.py
+cp resnet18.pt build
+
 cd build
 cmake .. -DTorch_DIR=$TORCH_PATH/share/cmake/Torch