From 36886a0521cc8b7d0b846f28aba7a3f252fb0d57 Mon Sep 17 00:00:00 2001 From: wjtest1215 Date: Thu, 20 Oct 2022 11:30:34 +0800 Subject: [PATCH] =?UTF-8?q?=E4=B8=8A=E4=BC=A0=E6=96=87=E4=BB=B6=E8=87=B3?= =?UTF-8?q?=20'npu'?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- npu/upload.py | 14 ++++++++++++++ npu/upload_for_c2net.py | 6 ++++++ 2 files changed, 20 insertions(+) create mode 100644 npu/upload.py create mode 100644 npu/upload_for_c2net.py 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/")