From 93dd98b67a67164ff4f4bf9df431eec9aaed9a02 Mon Sep 17 00:00:00 2001 From: liuzx Date: Fri, 14 Apr 2023 09:19:25 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20'pretrain=5Ffor=5Fc2net.py?= =?UTF-8?q?'?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- pretrain_for_c2net.py | 14 +------------- 1 file changed, 1 insertion(+), 13 deletions(-) diff --git a/pretrain_for_c2net.py b/pretrain_for_c2net.py index da79012..2c9d5e8 100644 --- a/pretrain_for_c2net.py +++ b/pretrain_for_c2net.py @@ -174,15 +174,6 @@ def DownloadModelFromQizhi(model_url, model_dir): if local_rank%8==0: C2netModelToEnv(model_url,model_dir) return -def UploadToQizhi(train_dir, obs_train_url): - device_num = int(os.getenv('RANK_SIZE')) - local_rank=int(os.getenv('RANK_ID')) - if device_num == 1: - EnvToObs(train_dir, obs_train_url) - if device_num > 1: - if local_rank%8==0: - EnvToObs(train_dir, obs_train_url) - return parser = argparse.ArgumentParser(description='MindSpore Lenet Example') ### --multi_data_url,--train_url,--device_target,These 3 parameters must be defined first in a multi-dataset, @@ -273,7 +264,4 @@ if __name__ == "__main__": model.train(epoch_size, ds_train, callbacks=[time_cb, ckpoint_cb, - LossMonitor()]) - ###Copy the trained output data from the local running environment back to obs, - ###and download it in the training task corresponding to the Qizhi platform - UploadToQizhi(train_dir,args.train_url) \ No newline at end of file + LossMonitor()]) \ No newline at end of file