Skip to content

GitLab

Commit 09b09eb5 authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

Enable NHNet trainer tests on mirrored strategies. 

PiperOrigin-RevId: 312515585
parent d0e6e751
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
official/nlp/nhnet/trainer_test.py
View file @ 09b09eb5
...@@ -37,6 +37,7 @@ def all_strategy_combinations(): ...@@ -37,6 +37,7 @@ def all_strategy_combinations():
distribution=[ distribution=[
strategy_combinations.one_device_strategy, strategy_combinations.one_device_strategy,
strategy_combinations.one_device_strategy_gpu, strategy_combinations.one_device_strategy_gpu,
strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
strategy_combinations.tpu_strategy, strategy_combinations.tpu_strategy,
], ],
mode="eager", mode="eager",
......
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