How to implement create_train_op in TF2?

TensorFlow General Discussion
,
1

I would like to use tf.contrib.training.create_train_op in TF2, as it was possible in TF1. However, since the contrib module is now deprecated in TF2, I need some kind of
workaround.

Thank you

3

Hi @camelfly, tf.contrib.layers are replaced with TF Slim symbols. In the 2.x version you use it with slim.learning.create_train_op( ). For more details please refer to this migration document. Thank You.

1 Like