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

Re-enable tests with old optimizers + dist strat + keras.

Browse Source 
PiperOrigin-RevId: 260619319
This commit is contained in:
Priya Gupta 2019-07-29 17:54:26 -07:00 committed by TensorFlower Gardener
parent e327064034
commit 5dd7dbb233
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/python/keras/distribute/distribute_strategy_test.py
View File

@ -274,6 +274,10 @@ def strategy_and_optimizer_combinations():
strategy_minus_tpu_combinations(),
combinations.combine(
optimizer=[
strategy_combinations.adagrad_optimizer_v1_fn,
strategy_combinations.adam_optimizer_v1_fn,
strategy_combinations.gradient_descent_optimizer_v1_fn,
strategy_combinations.rmsprop_optimizer_v1_fn,
strategy_combinations.adagrad_optimizer_keras_v2_fn,
strategy_combinations.adam_optimizer_keras_v2_fn,
strategy_combinations.gradient_descent_optimizer_keras_v2_fn,

2
tensorflow/python/keras/distribute/keras_embedding_model_correctness_test.py
View File

@ -51,8 +51,6 @@ class DistributionStrategyEmbeddingModelCorrectnessTest(
model.set_weights(initial_weights)
model.compile(
# TODO(b/130808953): Switch back the V1 optimizer once global_step is
# mirrored.
optimizer=gradient_descent_keras.SGD(learning_rate=0.1),
loss='sparse_categorical_crossentropy',
metrics=['sparse_categorical_accuracy'],