1
0
mirror of https://github.com/newnius/YAO-Dockerfiles.git synced 2025-06-06 07:51:54 +00:00
YAO-Dockerfiles/tensorflow/2.1-gpu/save.py

21 lines
516 B
Python

from hdfs import *
import os
import time
if __name__ == '__main__':
os.environ["TZ"] = 'Asia/Shanghai'
if hasattr(time, 'tzset'):
time.tzset()
try:
hdfs_address = os.environ['hdfs_address']
hdfs_dir = os.environ['hdfs_dir']
output_dir = os.environ['output_dir']
client = Client(hdfs_address)
client.upload(hdfs_dir, output_dir)
print('Save ' + output_dir + ' to' + hdfs_address + ' ' + hdfs_dir)
except Exception as e:
print('Unable to persist data to HDFS,', str(e))