I think there is a little mistakes in tony-example, as mentioned in https://github.com/linkedin/TonY/issues/492, because mnist_distributed.py is write in tensorflow 1.x, but in README, it said
Build a Python virtual environment with TensorFlow 2.3 installed
user should install tensorflow 1.x to run mnist_distributed.py
I think there is a little mistakes in tony-example, as mentioned in https://github.com/linkedin/TonY/issues/492, because mnist_distributed.py is write in tensorflow 1.x, but in README, it said
user should install tensorflow 1.x to run mnist_distributed.py