Browse Source

上传文件至 'npu'

test_v20221116
wjtest1215 1 year ago
parent
commit
36886a0521
2 changed files with 20 additions and 0 deletions
  1. +14
    -0
      npu/upload.py
  2. +6
    -0
      npu/upload_for_c2net.py

+ 14
- 0
npu/upload.py View File

@@ -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

+ 6
- 0
npu/upload_for_c2net.py View File

@@ -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/")

Loading…
Cancel
Save