saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-29 20:14:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Disable some unit tests for training (#6699)

Browse source
This commit is contained in:
Changming Sun 2021-02-16 09:45:59 -08:00 committed by GitHub
parent 33279250b5
commit 9a01174037
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 35 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
onnxruntime/test/providers/cpu/model_tests.cc
View file

@ -126,6 +126,40 @@ TEST_P(ModelTest, Run) {
{"momentum", "not a registered function/op", {}}, // Op not registered.
{"momentum_multiple", "not a registered function/op", {}}, // Op not registered.
{"nesterov_momentum", "not a registered function/op", {}}, // Op not registered.
{"softmax_cross_entropy_input_shape_is_NCd1d2d3d4d5_none_no_weight", "type error", {"onnx170"}},
{"softmax_cross_entropy_mean_weight_ignore_index_log_prob", "type error", {"onnx170"}},
{"softmax_cross_entropy_input_shape_is_NCd1_mean_weight_negative_ignore_index_log_prob", "type error", {"onnx170"}},
{"softmax_cross_entropy_mean_weight_log_prob", "type error", {"onnx170"}},
{"softmax_cross_entropy_input_shape_is_NCd1d2d3d4d5_mean_weight_log_prob", "type error", {"onnx170"}},
{"softmax_cross_entropy_mean_weight_ignore_index_3d", "type error", {"onnx170"}},
{"softmax_cross_entropy_mean_weight_ignore_index_4d_log_prob", "type error", {"onnx170"}},
{"softmax_cross_entropy_input_shape_is_NCd1d2d3d4d5_none_no_weight_log_prob", "type error", {"onnx170"}},
{"softmax_cross_entropy_mean_weight_ignore_index_4d", "type error", {"onnx170"}},
{"softmax_cross_entropy_mean", "type error", {"onnx170"}},
{"softmax_cross_entropy_mean_log_prob", "type error", {"onnx170"}},
{"softmax_cross_entropy_mean_no_weight_ignore_index", "type error", {"onnx170"}},
{"softmax_cross_entropy_input_shape_is_NCd1d2d3_sum_weight_high_ignore_index_log_prob", "type error", {"onnx170"}},
{"softmax_cross_entropy_mean_3d_log_prob", "type error", {"onnx170"}},
{"softmax_cross_entropy_none_log_prob", "type error", {"onnx170"}},
{"softmax_cross_entropy_mean_3d", "type error", {"onnx170"}},
{"softmax_cross_entropy_input_shape_is_NCd1d2d3_none_no_weight_negative_ignore_index", "type error", {"onnx170"}},
{"softmax_cross_entropy_mean_weight_ignore_index_3d_log_prob", "type error", {"onnx170"}},
{"softmax_cross_entropy_mean_no_weight_ignore_index_3d_log_prob", "type error", {"onnx170"}},
{"softmax_cross_entropy_none_weights_log_prob", "type error", {"onnx170"}},
{"softmax_cross_entropy_sum_log_prob", "type error", {"onnx170"}},
{"softmax_cross_entropy_mean_weight_ignore_index", "type error", {"onnx170"}},
{"softmax_cross_entropy_mean_no_weight_ignore_index_log_prob", "type error", {"onnx170"}},
{"softmax_cross_entropy_mean_no_weight_ignore_index_3d", "type error", {"onnx170"}},
{"softmax_cross_entropy_input_shape_is_NCd1d2d3_sum_weight_high_ignore_index", "type error", {"onnx170"}},
{"softmax_cross_entropy_sum", "type error", {"onnx170"}},
{"softmax_cross_entropy_input_shape_is_NCd1d2d3_none_no_weight_negative_ignore_index_log_prob", "type error", {"onnx170"}},
{"softmax_cross_entropy_none_weights", "type error", {"onnx170"}},
{"softmax_cross_entropy_mean_no_weight_ignore_index_4d_log_prob", "type error", {"onnx170"}},
{"softmax_cross_entropy_none", "type error", {"onnx170"}},
{"softmax_cross_entropy_input_shape_is_NCd1_mean_weight_negative_ignore_index", "type error", {"onnx170"}},
{"softmax_cross_entropy_input_shape_is_NCd1d2d3d4d5_mean_weight", "type error", {"onnx170"}},
{"softmax_cross_entropy_mean_weight", "type error", {"onnx170"}},
{"softmax_cross_entropy_mean_no_weight_ignore_index_4d", "type error", {"onnx170"}},
#endif
{"mask_rcnn_keras", "this model currently has an invalid contrib op version set to 10", {}}};

2
tools/ci_build/github/azure-pipelines/templates/py-packaging-stage.yml
View file

@ -270,7 +270,7 @@ stages:
timeoutInMinutes: 180
workspace:
clean: all
pool: Linux-GPU-CUDA10
pool: Onnxruntime-Linux-GPU
strategy:
matrix:
Python36: