diff --git a/train_continue.py b/train_continue.py index 63f2a75..99aaea7 100644 --- a/train_continue.py +++ b/train_continue.py @@ -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: