compat.v1.trainable_variables() (#6203)

3a19c394 · Toby Boyd · GitHub · ae699073 · 3a19c394
Unverified Commit 3a19c394 authored Feb 14, 2019 by Toby Boyd Committed by GitHub Feb 14, 2019
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/resnet/resnet_run_loop.py official/resnet/resnet_run_loop.py +1 -1

No files found.
--- a/official/resnet/resnet_run_loop.py
+++ b/official/resnet/resnet_run_loop.py
@@ -364,7 +364,7 @@ def resnet_model_fn(features, labels, mode, model_class,
      # loss is computed using fp32 for numerical stability.
      [
          tf.nn.l2_loss(tf.cast(v, tf.float32))
-          for v in tf.trainable_variables()
+          for v in tf.compat.v1.trainable_variables()
          if loss_filter_fn(v.name)
      ]) / tf.distribute.get_strategy().num_replicas_in_sync
  tf.compat.v1.summary.scalar('l2_loss', l2_loss)