update

README.md

@@ -103,11 +103,13 @@
 
   ```
   from openi import openi_multidataset_to_env as DatasetToEnv
+  DatasetToEnv(args.multi_data_url, data_dir)
   ```
 * 在智算集群导入数据集到训练环境使用c2net_multidataset_to_env
 
   ```
   from openi import c2net_multidataset_to_env as DatasetToEnv  
+  DatasetToEnv(args.multi_data_url, data_dir)
   ```
 
 ###### 结果输出部分：
@@ -116,8 +118,13 @@
 
   ```
   from openi import env_to_openi
+  env_to_openi(train_dir,args.train_url)
+  ```
+* 智算集群不需要使用任何函数，在训练结束后，只需要在代码中指定输出位置为/cache/output,系统会自动判别并将输出结果上传到启智平台
+* ```
+  train_dir = '/cache/output'
+  outputDirectory = train_dir 
   ```
-* 智算集群不需要使用任何函数，在训练结束后，系统会自动判别并将输出结果上传到启智平台
 
 ### 4.3. 训练环境与调试环境区别：
 

inference.py

@@ -29,14 +29,14 @@ from openi import pretrain_to_env
 parser = argparse.ArgumentParser(description='MindSpore Lenet Example')
 parser.add_argument('--multi_data_url',
                 type=str,
-                default= '/cache/data/',
+                default= '[{}]',
                 help='path where the dataset is saved')      
 parser.add_argument('--pretrain_url',
                 help='model to save/load',
-                default=  '/cache/checkpoint.ckpt')  
+                default=  '[{}]')  
 parser.add_argument('--result_url',
                 help='result folder to save/load',
-                default= '/cache/result/')   
+                default= '')   
 parser.add_argument('--device_target', type=str, default="Ascend", choices=['Ascend', 'GPU', 'CPU'],
                     help='device where the code will be implemented (default: Ascend)')                
 

pretrain_for_c2net.py

@@ -37,13 +37,7 @@ parser.add_argument('--multi_data_url',
                     default= '[{}]')                        
 parser.add_argument('--pretrain_url',
                 help='使用预训练模型单文件或多文件时，需要定义的参数',
-                default=  '[{}]')   
-parser.add_argument('--ckpt_url',
-                help='使用预训练模型单文件或多文件时，需要定义的参数',
-                default=  '[{}]')                  
-parser.add_argument('--train_url',
-                    help='回传结果到启智，需要定义的参数',
-                    default= '')  
+                default=  '[{}]')                 
 
 parser.add_argument(
     '--device_target',
@@ -72,7 +66,7 @@ if __name__ == "__main__":
     ###拷贝数据集到训练镜像
     DatasetToEnv(args.multi_data_url, data_dir)
     ###拷贝多个预训练模型文件到训练环境
-    pretrain_to_env(args.ckpt_url, pretrain_dir)
+    pretrain_to_env(args.pretrain_url, pretrain_dir)
 
     device_num = int(os.getenv('RANK_SIZE'))
     #使用单卡时

train_for_c2net.py

@@ -33,10 +33,6 @@ parser.add_argument('--multi_data_url',
                     help='使用数据集，需要定义的参数',
                     default= '[{}]')                        
 
-parser.add_argument('--train_url',
-                    help='回传结果到启智，需要定义的参数',
-                    default= '')  
-
 parser.add_argument(
     '--device_target',
     type=str,