更新 'train_continue.py'
This commit is contained in:
parent
09fddff670
commit
3da87c879e
|
|
@ -70,7 +70,7 @@ if __name__ == "__main__":
|
|||
except Exception as e:
|
||||
print("path already exists")
|
||||
|
||||
oponi_multidataset_to_Env(args.multi_data_url, data_dir)
|
||||
openi_multidataset_to_env(args.multi_data_url, data_dir)
|
||||
|
||||
device_num = int(os.getenv('RANK_SIZE'))
|
||||
if device_num == 1:
|
||||
|
|
|
|||
Loading…
Reference in New Issue