Does anyone get the Tensorflow distributed training notebook to work?

Question

When I run this notebook, which is linked from Tensorflows documentation site,

https://colab.research.google.com/github/tensorflow/docs/blob/r2.0rc/site/en/tutorials/distribute/training_loops.ipynb#scrollTo=7x7s5iYAYSGD

I get the following error:

module 'tensorflow._api.v1.nn' has no attribute 'compute_average_loss'

If anyone can get it to work, what settings are you using in Google Colab? Python 3/2, GPU, Tensorflow version, etc.

Thanks

Answer 1

The github code is from may this year:

https://github.com/tensorflow/docs/commits/r2.0rc/site/en/tutorials/distribute/training_loops.ipynb

while the colab is loading tf 1.14:

from __future__ import absolute_import, division, print_function, unicode_literals

# Import TensorFlow
import tensorflow as tf

# Helper libraries
import numpy as np
import os

print(tf.__version__)

1.14.0

so your best bet is to run the program with tf 2.0.

hope it helps:)