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

Fix the timeout of nasnet test.

Browse Source 
PiperOrigin-RevId: 294782893
Change-Id: Iffc97ba7ecb2072fd6d42ba7d9923952b157452d
This commit is contained in:
Yanhui Liang 2020-02-12 16:15:52 -08:00 committed by TensorFlower Gardener
parent d74394747a
commit 13655728cd
2 changed files with 20 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -235,14 +235,29 @@ tf_py_test(
)
tf_py_test(
name = "applications_load_weight_test_nasnet",
name = "applications_load_weight_test_nasnet_mobile",
srcs = ["applications_load_weight_test.py"],
args = ["--module=nasnet"],
args = ["--module=nasnet_mobile"],
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_nasnet_large",
srcs = ["applications_load_weight_test.py"],
args = ["--module=nasnet_large"],
main = "applications_load_weight_test.py",
tags = [
"no_oss",
"no_pip",
"notsan", # b/149100891
],
deps = [
":applications",

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

@ -53,7 +53,8 @@ ARG_TO_MODEL = {
'mobilenet_v2': (mobilenet_v2, [mobilenet_v2.MobileNetV2]),
'densenet': (densenet, [densenet.DenseNet121,
densenet.DenseNet169, densenet.DenseNet201]),
'nasnet': (nasnet, [nasnet.NASNetMobile, nasnet.NASNetLarge]),
'nasnet_mobile': (nasnet, [nasnet.NASNetMobile]),
'nasnet_large': (nasnet, [nasnet.NASNetLarge]),
'efficientnet': (efficientnet,
[efficientnet.EfficientNetB0, efficientnet.EfficientNetB1,
efficientnet.EfficientNetB2, efficientnet.EfficientNetB3,