Skip to content

Commit 07e9e0f

Browse files
authored
Keras nightly CI jobs update. (#1904)
* Update the tests scripts for all keras2onnx_application_test on Windows. Signed-off-by: Jay Zhang <[email protected]>
1 parent de7e5af commit 07e9e0f

File tree

2 files changed

+37
-12
lines changed

2 files changed

+37
-12
lines changed

ci_build/azure_pipelines/keras2onnx_application_tests.yml

Lines changed: 36 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@ jobs:
1717
INSTALL_ORT: pip install onnxruntime==1.8.0
1818
INSTALL_KERAS_RESNET: pip install keras-resnet
1919
INSTALL_TRANSFORMERS:
20+
INSTALL_NUMPY:
2021
NIGHTLY_BUILD_TEST: python run_all.py --exclude "test_keras_applications_v2.py"
2122

2223
Python37-onnx1.6:
@@ -40,17 +41,29 @@ jobs:
4041
INSTALL_ORT: pip install onnxruntime==1.9.0
4142
INSTALL_KERAS_RESNET: pip install keras-resnet
4243
INSTALL_TRANSFORMERS:
43-
INSTALL_NUMPY: pip install numpy==1.19.0
44+
INSTALL_NUMPY:
4445
NIGHTLY_BUILD_TEST: python run_all.py --exclude "test_keras_applications_v2.py"
4546

46-
Python37-tf2.x:
47+
Python37-onnx1.11:
4748
python.version: '3.7'
4849
ONNX_PATH: onnx==1.11.0
49-
INSTALL_KERAS:
50-
UNINSTALL_KERAS: pip uninstall keras -y
51-
INSTALL_TENSORFLOW: pip install tensorflow==2.2.0
50+
INSTALL_KERAS: pip install keras==2.3.1
51+
UNINSTALL_KERAS:
52+
INSTALL_TENSORFLOW: pip install tensorflow==1.15.0
5253
INSTALL_ORT: pip install onnxruntime==1.11.0
5354
INSTALL_KERAS_RESNET: pip install keras-resnet
55+
INSTALL_TRANSFORMERS:
56+
INSTALL_NUMPY: pip install numpy==1.19.0
57+
NIGHTLY_BUILD_TEST: python run_all_v2.py --exclude "test_keras_applications_v2.py"
58+
59+
Python38-tf2.x:
60+
python.version: '3.8'
61+
ONNX_PATH: onnx==1.11.0
62+
INSTALL_KERAS:
63+
UNINSTALL_KERAS: pip uninstall keras -y
64+
INSTALL_TENSORFLOW: pip install tensorflow==2.5.0
65+
INSTALL_ORT: pip install onnxruntime==1.9.0
66+
INSTALL_KERAS_RESNET: pip install keras-resnet
5467
INSTALL_TRANSFORMERS: pip install transformers==3.4.0
5568
INSTALL_NUMPY: pip install numpy==1.19.0
5669
NIGHTLY_BUILD_TEST: python run_all_v2.py
@@ -75,7 +88,7 @@ jobs:
7588
INSTALL_ORT: pip install onnxruntime==1.8.0
7689
INSTALL_KERAS_RESNET: pip install keras-resnet
7790
INSTALL_TRANSFORMERS:
78-
NIGHTLY_BUILD_TEST: python run_all.py --exclude "test_keras_applications_v2.py test_mask_rcnn.py"
91+
NIGHTLY_BUILD_TEST: python run_all_v2.py --exclude "test_keras_applications_v2.py"
7992

8093
Python37-onnx1.6:
8194
python.version: '3.7'
@@ -87,7 +100,7 @@ jobs:
87100
INSTALL_KERAS_RESNET: pip install keras-resnet
88101
INSTALL_TRANSFORMERS:
89102
INSTALL_NUMPY: pip install numpy==1.19.0
90-
NIGHTLY_BUILD_TEST: python run_all.py --exclude "test_keras_applications_v2.py test_mask_rcnn.py"
103+
NIGHTLY_BUILD_TEST: python run_all_v2.py --exclude "test_keras_applications_v2.py"
91104

92105
Python37-onnx1.9:
93106
python.version: '3.7'
@@ -99,19 +112,31 @@ jobs:
99112
INSTALL_KERAS_RESNET: pip install keras-resnet
100113
INSTALL_TRANSFORMERS:
101114
INSTALL_NUMPY: pip install numpy==1.19.0
102-
NIGHTLY_BUILD_TEST: python run_all.py --exclude "test_keras_applications_v2.py"
115+
NIGHTLY_BUILD_TEST: python run_all_v2.py --exclude "test_keras_applications_v2.py"
116+
117+
Python37-onnx1.11:
118+
python.version: '3.7'
119+
ONNX_PATH: onnx==1.11.0
120+
INSTALL_KERAS: pip install keras==2.3.1
121+
UNINSTALL_KERAS:
122+
INSTALL_TENSORFLOW: pip install tensorflow==1.15.0
123+
INSTALL_ORT: pip install onnxruntime==1.11.0
124+
INSTALL_KERAS_RESNET: pip install keras-resnet
125+
INSTALL_TRANSFORMERS:
126+
INSTALL_NUMPY: pip install numpy==1.19.0
127+
NIGHTLY_BUILD_TEST: python run_all_v2.py --exclude "test_keras_applications_v2.py"
103128

104129
Python38-tf2.x:
105130
python.version: '3.8'
106131
ONNX_PATH: onnx==1.11.0
107132
INSTALL_KERAS:
108133
UNINSTALL_KERAS: pip uninstall keras -y
109-
INSTALL_TENSORFLOW: pip install tensorflow==2.2.0
110-
INSTALL_ORT: pip install onnxruntime==1.11.0
134+
INSTALL_TENSORFLOW: pip install tensorflow==2.5.0
135+
INSTALL_ORT: pip install onnxruntime==1.9.0
111136
INSTALL_KERAS_RESNET: pip install keras-resnet
112137
INSTALL_TRANSFORMERS: pip install transformers==3.4.0
113138
INSTALL_NUMPY: pip install numpy==1.19.0
114-
NIGHTLY_BUILD_TEST: python run_all_v2.py --exclude "test_keras_applications_v2.py test_mask_rcnn.py"
139+
NIGHTLY_BUILD_TEST: python run_all_v2.py --exclude "test_keras_applications_v2.py"
115140

116141
steps:
117142
- template: 'templates/keras2onnx_application_tests.yml'

ci_build/azure_pipelines/trimmed_keras2onnx_application_tests.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ jobs:
88
vmImage: 'ubuntu-latest'
99
strategy:
1010
matrix:
11-
Python36-onnx1.5:
11+
Python36-onnx1.10:
1212
python.version: '3.6'
1313
ONNX_PATH: onnx==1.10.2
1414
INSTALL_KERAS: pip install keras==2.2.4

0 commit comments

Comments
 (0)