diff --git a/npu/upload.py b/npu/upload.py new file mode 100644 index 0000000..6060ab2 --- /dev/null +++ b/npu/upload.py @@ -0,0 +1,14 @@ +from mindspore.train.callback import Callback +import moxing as mox + +class UploadOutput(Callback): + def __init__(self, train_dir, obs_train_url): + self.train_dir = train_dir + self.obs_train_url = obs_train_url + def epoch_end(self,run_context): + try: + mox.file.copy_parallel(self.train_dir , self.obs_train_url ) + print("Successfully Upload {} to {}".format(self.train_dir ,self.obs_train_url )) + except Exception as e: + print('moxing upload {} to {} failed: '.format(self.train_dir ,self.obs_train_url ) + str(e)) + return diff --git a/npu/upload_for_c2net.py b/npu/upload_for_c2net.py new file mode 100644 index 0000000..b725724 --- /dev/null +++ b/npu/upload_for_c2net.py @@ -0,0 +1,6 @@ +from mindspore.train.callback import Callback +import os + +class UploadOutput(Callback): + def epoch_end(self,run_context): + os.system("cd /cache/script_for_grampus/ &&./uploader_for_npu " + "/cache/output/")