更新 'train_for_multidataset_dataparallel.py'

2022-08-04 09:19:02 +08:00 · 2022-08-04 09:19:02 +08:00 · dd8b498193
parent 189a3aed03
commit dd8b498193
1 changed files with 249 additions and 249 deletions
--- a/train_for_multidataset_dataparallel.py
+++ b/train_for_multidataset_dataparallel.py
@ -72,7 +72,7 @@ def MultiObsToEnv(multi_data_url, workroot):
    The path to the checkpoint_lenet-1_1875 dataset is /home/work/user-job-dir/checkpoint_lenet-1_1875

 (3)Copy the output model to obs.
-def EnvToObs(obs_train_url, train_dir):
+def EnvToObs(train_dir, obs_train_url):
    try:
        mox.file.copy_parallel(train_dir, obs_train_url)
        print("Successfully Upload {} to {}".format(train_dir,