train.py 文件源码

def redirect_stdout(target):
        original = sys.stdout
        sys.stdout = target
        yield
        sys.stdout = original

# To deactivate warnings: https://github.com/tensorflow/tensorflow/issues/7778