mesh
mesh copied to clipboard
tf2 in mesh_tensorflow/utils.py incompatible with tensor2tensor/rl
I'm wondering if tf2 is absolutely needed in mesh_tensorflow/utils.py? I'm trying to reproduce on the provided Google colab https://github.com/tensorflow/tensor2tensor/tree/master/tensor2tensor/rl
with tensorflow 1.13.1 and T2T 1.13.1 (the recommended config), but I got stuck at line 26 import tensorflow.compat.v2 as tf2
because I'm using tensorflow 1.13.1
Would it be possible to make mesh_tensorflow compatible with tensorflow v1?