Skip to content

Commit 80d6080

Browse files
Trim down regular keras2onnx tests (#1618)
Signed-off-by: Tom Wildenhain <[email protected]> Co-authored-by: Guenther Schmuelling <[email protected]>
1 parent 11983f8 commit 80d6080

File tree

4 files changed

+112
-5
lines changed

4 files changed

+112
-5
lines changed

ci_build/azure_pipelines/keras2onnx_application_tests.yml

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,4 @@
1-
# Python package
2-
# Create and test a Python package on multiple Python versions.
3-
# Add steps that analyze code, save the dist with the build record, publish to a PyPI-compatible index, and more:
4-
# https://docs.microsoft.com/azure/devops/pipelines/languages/python
1+
# Nightly keras2onnx application tests
52

63
jobs:
74

ci_build/azure_pipelines/keras2onnx_unit_test.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# Tests copied from keras2onnx
1+
# Nightly keras2onnx unit tests
22

33
jobs:
44

Lines changed: 36 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,36 @@
1+
# keras2onnx application tests (trimmed)
2+
3+
jobs:
4+
5+
- job: 'Linux'
6+
timeoutInMinutes: 180
7+
pool:
8+
vmImage: 'Ubuntu-16.04'
9+
strategy:
10+
matrix:
11+
Python36-onnx1.5:
12+
python.version: '3.6'
13+
ONNX_PATH: onnx==1.5.0
14+
INSTALL_KERAS: pip install keras==2.2.4
15+
UNINSTALL_KERAS:
16+
INSTALL_TENSORFLOW: pip install tensorflow==1.15.0
17+
INSTALL_ORT: pip install onnxruntime==1.8.0
18+
INSTALL_KERAS_RESNET: pip install keras-resnet
19+
INSTALL_TRANSFORMERS:
20+
NIGHTLY_BUILD_TEST: python run_all.py --exclude "test_keras_applications_v2.py"
21+
22+
Python38-tf2:
23+
python.version: '3.8'
24+
ONNX_PATH: onnx==1.9.0
25+
INSTALL_KERAS:
26+
UNINSTALL_KERAS: pip uninstall keras -y
27+
INSTALL_TENSORFLOW: pip install tensorflow==2.2.0
28+
INSTALL_ORT: pip install onnxruntime==1.8.0
29+
INSTALL_KERAS_RESNET: pip install keras-resnet
30+
INSTALL_TRANSFORMERS: pip install transformers==3.4.0
31+
NIGHTLY_BUILD_TEST: python run_all_v2.py
32+
33+
steps:
34+
- template: 'templates/keras2onnx_application_tests.yml'
35+
parameters:
36+
platform: 'linux'
Lines changed: 74 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,74 @@
1+
# Tests copied from keras2onnx
2+
3+
jobs:
4+
5+
- job: 'Linux'
6+
pool:
7+
vmImage: 'Ubuntu-16.04'
8+
strategy:
9+
matrix:
10+
11+
############ TF Keras Unit Tests ############
12+
Python36-tf1.15:
13+
python.version: '3.6'
14+
ONNX_PATH: onnx==1.5.0
15+
TENSORFLOW_PATH: tensorflow==1.15.0
16+
INSTALL_ORT: pip install onnxruntime==1.8.0
17+
18+
Python38-tf2.5:
19+
python.version: '3.8'
20+
ONNX_PATH: onnx==1.8.0
21+
TENSORFLOW_PATH: tensorflow-cpu==2.5.0
22+
INSTALL_ORT: pip install onnxruntime==1.8.0
23+
24+
############ Pure Keras Unit Tests ############
25+
Keras-Py37-tf1.15.0:
26+
python.version: '3.7'
27+
ONNX_PATH: onnx==1.9.0
28+
KERAS: keras==2.4.3
29+
TENSORFLOW_PATH: tensorflow==1.15.0
30+
INSTALL_ORT: pip install onnxruntime==1.8.0
31+
32+
# UT for keras 2.3 need tensorflow <= 2.0.0
33+
Keras-Py37-tf2.0.0:
34+
python.version: '3.7'
35+
ONNX_PATH: onnx==1.6.0
36+
KERAS: keras==2.3.1
37+
TENSORFLOW_PATH: tensorflow==2.0.0
38+
INSTALL_ORT: pip install onnxruntime==1.8.0
39+
40+
steps:
41+
- template: 'templates/keras2onnx_unit_test.yml'
42+
parameters:
43+
platform: 'linux'
44+
45+
- job: 'Win'
46+
pool:
47+
vmImage: 'vs2017-win2016'
48+
strategy:
49+
matrix:
50+
############ TF Keras Unit Tests ############
51+
Python36-onnx1.2:
52+
python.version: '3.6'
53+
ONNX_PATH: onnx==1.2.3
54+
TENSORFLOW_PATH: tensorflow==1.15.0
55+
INSTALL_ORT: pip install onnxruntime==1.8.0
56+
57+
Python37-tf2.3:
58+
python.version: '3.7'
59+
ONNX_PATH: onnx==1.9.0
60+
TENSORFLOW_PATH: tensorflow-cpu==2.3.0
61+
INSTALL_ORT: pip install onnxruntime==1.8.0
62+
63+
############ Pure Keras Unit Tests ############
64+
Keras-Py37-tf2.2.0:
65+
python.version: '3.7'
66+
ONNX_PATH: onnx==1.9.0
67+
KERAS: keras==2.4.3
68+
TENSORFLOW_PATH: tensorflow==2.2.0
69+
INSTALL_ORT: pip install onnxruntime==1.8.0
70+
71+
steps:
72+
- template: 'templates/keras2onnx_unit_test.yml'
73+
parameters:
74+
platform: 'windows'

0 commit comments

Comments
 (0)