Transformer implemented with tensorflow and horovod, but is failed due to "GetNext() failed because the iterator has not been initialized. Ensure that you have run the init