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

Set use_gpu=True in _testTypesForSparseAdagrad

Browse Source
This commit is contained in:
Ben Barsdell 2021-01-08 16:52:04 +11:00
parent 28162ecc0c
commit d0ebdb98e1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/python/training/training_ops_test.py
View File

@ -225,7 +225,7 @@ class TrainingOpsTest(TensorFlowTestCase):
def _testTypesForSparseAdagrad(self, x, y, lr, grad, indices): def _testTypesForSparseAdagrad(self, x, y, lr, grad, indices):
self.setUp() self.setUp()
with self.session(use_gpu=False): with self.session(use_gpu=True):
var = variables.VariableV1(x) var = variables.VariableV1(x)
accum = variables.VariableV1(y) accum = variables.VariableV1(y)
self.evaluate(variables.global_variables_initializer()) self.evaluate(variables.global_variables_initializer())