dreiss

diff --git i/test/test_nnapi.py w/test/test_nnapi.py
index 23165c1d78..4328a73112 100644
--- i/test/test_nnapi.py
+++ w/test/test_nnapi.py
@@ -68,6 +68,7 @@ class TestNNAPI(TestCase):
                     # Too many mismatches.  Re-run the check with no tolerance
                     # to get a nice message.
                     self.assertEqual(eager_output, nnapi_output, atol=0, rtol=0)
+            return eager_output, nnapi_output
 

dreiss

diff --git i/test/test_nnapi.py w/test/test_nnapi.py
index 23165c1d78..4328a73112 100644
--- i/test/test_nnapi.py
+++ w/test/test_nnapi.py
@@ -68,6 +68,7 @@ class TestNNAPI(TestCase):
                     # Too many mismatches.  Re-run the check with no tolerance
                     # to get a nice message.
                     self.assertEqual(eager_output, nnapi_output, atol=0, rtol=0)
+            return eager_output, nnapi_output
 

dreiss

diff --git i/test/test_nnapi.py w/test/test_nnapi.py
index 23165c1d78..4328a73112 100644
--- i/test/test_nnapi.py
+++ w/test/test_nnapi.py
@@ -68,6 +68,7 @@ class TestNNAPI(TestCase):
                     # Too many mismatches.  Re-run the check with no tolerance
                     # to get a nice message.
                     self.assertEqual(eager_output, nnapi_output, atol=0, rtol=0)
+            return eager_output, nnapi_output
 

dreiss

---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-13-c9831efca396> in <module>
----> 1 torch.onnx.export(Model(), torch.zeros(1,1,1,1), MODEL_FILE, opset_version=11)
      2 
      3 loaded = onnx.load(MODEL_FILE)

~/work/pytorch/torch/onnx/__init__.py in export(model, args, f, export_params, verbose, training, input_names, output_names, aten, export_raw_ir, operator_export_type, opset_version, _retain_param_name, do_constant_folding, example_outputs, strip_doc_string, dynamic_axes, keep_initializers_as_inputs, custom_opsets, enable_onnx_checker, use_external_data_format)
    170                         do_constant_folding, example_outputs,
    171                         strip_doc_string, dynamic_axes, keep_initializers_as_inputs,

dreiss

# Get the aarch64 Minimal image from https://fedoraproject.org/wiki/Architectures/ARM/Raspberry_Pi#aarch64_supported_images_for_Raspberry_Pi_3
# Log in as root.  Create new user.  Log in as user.
sudo dnf update
sudo dnf install git cmake ninja-build clang unzip distcc python3-virtualenv python3-devel python3-opencv gstreamer1-plugins-base gstreamer1-plugins-good
sudo reboot
# Set up distcc symlinks for PATH.
mkdir ~/distcc-symlinks
mkdir ln -s `which distcc` ~/distcc-symlinks/clang
mkdir ln -s `which distcc` ~/distcc-symlinks/clang++
# NOTE: --system-site-packages is to get access to OpenCV from system package

dreiss

diff --git c/fb_sol_v1/ocr_lib.py w/fb_sol_v1/ocr_lib.py
index 7f71e92..cd72b0f 100644
--- c/fb_sol_v1/ocr_lib.py
+++ w/fb_sol_v1/ocr_lib.py
@@ -32,14 +32,17 @@ class OCRLib:
         )
 
     def process_rgb_image(self, input_image):
-        boxes, scores = self.detector.detect(input_image)
+        with torch.autograd.profiler.record_function("detect"):

dreiss

dreiss

#!/bin/sh
set -x
set -e
PT_JAVA_HOME=/usr/lib/jvm/java-1.8.0-openjdk-1.8.0.232.b09-0.fc30.x86_64
PT_LIBTORCH_HOME=$HOME/opt/libtorch
NUM_COMPILE_THREADS=8

cd android/pytorch_android/host
cmake -S .. -B build-cmake -DJAVA_HOME=$PT_JAVA_HOME -DLIBTORCH_HOME=$PT_LIBTORCH_HOME
cmake --build build-cmake -j $NUM_COMPILE_THREADS

dreiss

diff --git i/android/settings.gradle w/android/settings.gradle
index 99e442b2ae..1c63b71092 100644
--- i/android/settings.gradle
+++ w/android/settings.gradle
@@ -1,6 +1,2 @@
-include ':app', ':pytorch_android', ':fbjni', ':pytorch_android_torchvision', ':pytorch_host'
-
-project(':fbjni').projectDir = file('libs/fbjni_local')
-project(':pytorch_android_torchvision').projectDir = file('pytorch_android_torchvision')
-

dreiss