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

Use default_strategy instead of dummy scope

Browse Source
This commit is contained in:
Lukas Geiger 2020-06-16 02:25:18 +01:00
parent fe6580b4d8
commit da2046b8a3
1 changed files with 1 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tensorflow/python/keras/mixed_precision/experimental/autocast_variable_test.py
View File

@ -17,7 +17,6 @@ from __future__ import absolute_import
from __future__ import division from __future__ import division
from __future__ import print_function from __future__ import print_function
import contextlib
import os import os
from absl.testing import parameterized from absl.testing import parameterized
@ -43,15 +42,9 @@ from tensorflow.python.platform import test
from tensorflow.python.training import gradient_descent as gradient_descent_v1 from tensorflow.python.training import gradient_descent as gradient_descent_v1
from tensorflow.python.training.tracking import util as trackable_utils from tensorflow.python.training.tracking import util as trackable_utils
class DummyStrategy(object):
@contextlib.contextmanager
def scope(self):
yield
maybe_distribute = combinations.combine( maybe_distribute = combinations.combine(
distribution=[ distribution=[
combinations.NamedDistribution( strategy_combinations.default_strategy,
"Dummy", lambda: DummyStrategy(), required_gpus=None),
strategy_combinations.mirrored_strategy_with_cpu_1_and_2 strategy_combinations.mirrored_strategy_with_cpu_1_and_2
]) ])