From 8e53d25ab1c702dd6a0428aa2bb1f3d9788102cb Mon Sep 17 00:00:00 2001 From: czzhangheng Date: Tue, 11 Nov 2025 18:59:12 +0800 Subject: [PATCH] =?UTF-8?q?=E5=B0=86=E6=A3=80=E6=B5=8B=E6=95=B0=E6=8D=AE?= =?UTF-8?q?=E9=9B=86=E9=80=BB=E8=BE=91=E6=8C=AA=E5=88=B0=E9=A1=B6=E9=83=A8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- run.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/run.py b/run.py index 5785dbf..0a4c49a 100755 --- a/run.py +++ b/run.py @@ -1,5 +1,7 @@ import torch - +from utils.Download_data import check_and_download_data +data_complete = check_and_download_data() +assert data_complete is not None, "数据集下载失败，请重试！" # import time from config.args_parser import parse_args @@ -58,10 +60,5 @@ def main(): case _: raise ValueError(f"Unsupported mode: {args['basic']['mode']}") - if __name__ == "__main__": - from utils.Download_data import check_and_download_data - - data_complete = check_and_download_data() - assert data_complete is not None, "数据集下载失败，请重试！" main()