This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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 | |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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 | |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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 | |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
--------------------------------------------------------------------------- | |
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, |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
# 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 |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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"): |
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#!/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 |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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') | |
- |
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
NewerOlder