experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Split the application test for mobilenet_v3 into 2 to reduce the chance of timeout.

Browse Source 
PiperOrigin-RevId: 332894944
Change-Id: Iff0da41af951bb43a19244347eabb06e07a7c2a6
This commit is contained in:
Scott Zhu 2020-09-21 11:31:32 -07:00 committed by TensorFlower Gardener
parent 4c425d4dbf
commit a09165ad84
2 changed files with 20 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
tensorflow/python/keras/applications/BUILD
View File

@ -221,14 +221,29 @@ tf_py_test(
)
tf_py_test(
name = "applications_load_weight_test_mobilenet_v3",
name = "applications_load_weight_test_mobilenet_v3_small",
srcs = ["applications_load_weight_test.py"],
args = ["--module=mobilenet_v3"],
args = ["--module=mobilenet_v3_small"],
main = "applications_load_weight_test.py",
tags = [
"no_oss",
"no_pip",
],
deps = [
":applications",
"//tensorflow/python:client_testlib",
"@absl_py//absl/testing:parameterized",
],
)
tf_py_test(
name = "applications_load_weight_test_mobilenet_v3_large",
srcs = ["applications_load_weight_test.py"],
args = ["--module=mobilenet_v3_large"],
main = "applications_load_weight_test.py",
tags = [
"no_oss",
"no_pip",
"notsan",
],
deps = [
":applications",

4
tensorflow/python/keras/applications/applications_load_weight_test.py
View File

@ -52,8 +52,8 @@ ARG_TO_MODEL = {
[inception_resnet_v2.InceptionResNetV2]),
'mobilenet': (mobilenet, [mobilenet.MobileNet]),
'mobilenet_v2': (mobilenet_v2, [mobilenet_v2.MobileNetV2]),
'mobilenet_v3': (mobilenet_v3, [mobilenet_v3.MobileNetV3Small,
mobilenet_v3.MobileNetV3Large]),
'mobilenet_v3_small': (mobilenet_v3, [mobilenet_v3.MobileNetV3Small]),
'mobilenet_v3_large': (mobilenet_v3, [mobilenet_v3.MobileNetV3Large]),
'densenet': (densenet, [densenet.DenseNet121,
densenet.DenseNet169, densenet.DenseNet201]),
'nasnet_mobile': (nasnet, [nasnet.NASNetMobile]),