Skip to content

GitLab

Commit 241c985a authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

Img cls trainer tests on mirrored strategies. 

PiperOrigin-RevId: 312560339
parent 09b09eb5
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
official/vision/image_classification/classifier_trainer_test.py
View file @ 241c985a
...@@ -50,6 +50,7 @@ def distribution_strategy_combinations() -> Iterable[Tuple[Any, ...]]: ...@@ -50,6 +50,7 @@ def distribution_strategy_combinations() -> Iterable[Tuple[Any, ...]]:
strategy_combinations.default_strategy, strategy_combinations.default_strategy,
strategy_combinations.tpu_strategy, strategy_combinations.tpu_strategy,
strategy_combinations.one_device_strategy_gpu, strategy_combinations.one_device_strategy_gpu,
strategy_combinations.mirrored_strategy_with_two_gpus,
], ],
model=[ model=[
'efficientnet', 'efficientnet',
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment