Increase shard count for keras_embedding_model_correctness_test.
PiperOrigin-RevId: 327538956 Change-Id: I1cc98e6e40e507500758230341440d90bc6d8d85
This commit is contained in:
parent
e2ff54f938
commit
8257866ced
@ -417,7 +417,7 @@ distribute_py_test(
|
||||
srcs = ["keras_embedding_model_correctness_test.py"],
|
||||
full_precision = True,
|
||||
main = "keras_embedding_model_correctness_test.py",
|
||||
shard_count = 4,
|
||||
shard_count = 8,
|
||||
tags = [
|
||||
"multi_and_single_gpu",
|
||||
"no_windows_gpu",
|
||||
|
Loading…
Reference in New Issue
Block a user