From 442822b3fd6ea515d2d808c6f8ae95c167cd3dbc Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=BC=A0=E4=B9=BE=E4=B9=BE?= Date: Mon, 18 Aug 2025 16:39:13 +0800 Subject: [PATCH] opendwm_readme_fix --- model_examples/OpenDWM/OpenDWM.patch | 14 ++++---- model_examples/OpenDWM/README.md | 32 +++++++++---------- model_examples/OpenDWM/test/train.sh | 2 -- .../OpenDWM/test/train_performance.sh | 2 -- 4 files changed, 22 insertions(+), 28 deletions(-) diff --git a/model_examples/OpenDWM/OpenDWM.patch b/model_examples/OpenDWM/OpenDWM.patch index aa6437a7..b4fefe67 100644 --- a/model_examples/OpenDWM/OpenDWM.patch +++ b/model_examples/OpenDWM/OpenDWM.patch @@ -7,7 +7,7 @@ index 41eaad2..910a69d 100644 "fs": { "_class_name": "dwm.fs.dirfs.DirFileSystem", - "path": "/cache/aoss-v2.st-sh-01.sensecoreapi-oss.cn/users/wuzehuan" -+ "path": "data/nuscences" ++ "path": "data/nuscenes" }, "paths": [ - "workspaces/worldmodels/data/nuscenes/interp_12Hz_trainval.zip", @@ -71,8 +71,8 @@ index 41eaad2..910a69d 100644 "image_description_settings": { - "path": "/cache/aoss-v2.st-sh-01.sensecoreapi-oss.cn/users/wuzehuan/workspaces/worldmodels/data/nuscenes_v1.0-trainval_caption_v2_train.json", - "time_list_dict_path": "/cache/aoss-v2.st-sh-01.sensecoreapi-oss.cn/users/wuzehuan/workspaces/worldmodels/data/nuscenes_v1.0-trainval_caption_v2_times_train.json", -+ "path": "data/nuscences/nuscenes_v1.0-trainval_caption_v2_train.json", -+ "time_list_dict_path": "data/nuscences/nuscenes_v1.0-trainval_caption_v2_times_train.json", ++ "path": "data/nuscenes/nuscenes_v1.0-trainval_caption_v2_train.json", ++ "time_list_dict_path": "data/nuscenes/nuscenes_v1.0-trainval_caption_v2_times_train.json", "align_keys": [ "time", "weather" @@ -82,8 +82,8 @@ index 41eaad2..910a69d 100644 "image_description_settings": { - "path": "/cache/aoss-v2.st-sh-01.sensecoreapi-oss.cn/users/wuzehuan/workspaces/worldmodels/data/nuscenes_v1.0-trainval_caption_v2_val.json", - "time_list_dict_path": "/cache/aoss-v2.st-sh-01.sensecoreapi-oss.cn/users/wuzehuan/workspaces/worldmodels/data/nuscenes_v1.0-trainval_caption_v2_times_val.json", -+ "path": "data/nuscences/nuscenes_v1.0-trainval_caption_v2_val.json", -+ "time_list_dict_path": "data/nuscences/nuscenes_v1.0-trainval_caption_v2_times_val.json", ++ "path": "data/nuscenes/nuscenes_v1.0-trainval_caption_v2_val.json", ++ "time_list_dict_path": "data/nuscenes/nuscenes_v1.0-trainval_caption_v2_times_val.json", "align_keys": [ "time", "weather" @@ -321,7 +321,7 @@ index 1e74e10..3896cba 100644 + # 打印该样本的统计摘要 + print(f"\nSample {i_id} inference stats:") + print(f" Runs: {args.repeat}") -+ print(f" Mean time: {sample_stats['mean_time']:.4f} ± {sample_stats['std_time']:.4f} seconds") ++ print(f" Mean time: {sample_stats['mean_time']:.4f} seconds") + print(f" Min time: {sample_stats['min_time']:.4f} seconds") + print(f" Max time: {sample_stats['max_time']:.4f} seconds") + @@ -357,7 +357,7 @@ index 1e74e10..3896cba 100644 + print(f"Total samples processed: {global_stats['total_samples']}") + print(f"Total inference runs: {global_stats['total_runs']}") + print(f"Total inference time: {global_stats['total_time']:.2f} seconds") -+ print(f"Average per run: {global_stats['avg_time']:.4f} ± {global_stats['std_time']:.4f} seconds") ++ print(f"Average per run: {global_stats['avg_time']:.4f} seconds") + print(f"Fastest run: {global_stats['min_time']:.4f} seconds") + print(f"Slowest run: {global_stats['max_time']:.4f} seconds") + print(f"Detailed stats saved to: {stats_path}") diff --git a/model_examples/OpenDWM/README.md b/model_examples/OpenDWM/README.md index 62b8bf23..69294c6b 100644 --- a/model_examples/OpenDWM/README.md +++ b/model_examples/OpenDWM/README.md @@ -84,15 +84,11 @@ OpenDWM是一种统一的多视角驾驶视频生成框架。通过融合单/多 source {cann_root_dir}/set_env.sh ``` -2. 安装其他依赖项 - ``` - pip install -r requirements.txt - ``` -3. 安装Driving SDK +2. 安装Driving SDK 请参考昇腾[Driving SDK](https://gitee.com/ascend/DrivingSDK)代码仓说明编译安装Driving SDK -4. 安装MindSpeed +3. 安装MindSpeed 源码安装： ``` @@ -100,7 +96,7 @@ OpenDWM是一种统一的多视角驾驶视频生成框架。通过融合单/多 pip install -e MindSpeed ``` -3. 克隆代码仓到当前目录： +4. 克隆代码仓到当前目录： ``` git clone https://gitee.com/ascend/DrivingSDK.git -b master @@ -112,7 +108,7 @@ OpenDWM是一种统一的多视角驾驶视频生成框架。通过融合单/多 将模型根目录记作 `model-root-path` -4. 安装模型相关的依赖项。 +5. 安装模型相关的依赖项。 ``` python -m pip install torchvision==0.20.1 @@ -126,7 +122,7 @@ OpenDWM是一种统一的多视角驾驶视频生成框架。通过融合单/多 pip install $ASCEND_CUSTOM_PATH/latest/lib64/hccl-*-py3-none-any.whl ``` -5. 使用 patch 文件： +6. 使用 patch 文件： ``` cp -f ../OpenDWM.patch . git apply --reject --whitespace=fix OpenDWM.patch @@ -138,7 +134,7 @@ OpenDWM是一种统一的多视角驾驶视频生成框架。通过融合单/多 - 根据原仓**Train**章节准备数据集 - 1. 下载[nuScences数据集](https://www.nuscenes.org/download)到${model-root-path}/data/nuscenes，目录结构如下 + 1. 下载[nuScenes数据集](https://www.nuscenes.org/download)到${model-root-path}/data/nuscenes，目录结构如下 ```bash ${model-root-path}/data/nuscenes @@ -221,14 +217,14 @@ ${model-root-path}/pretrained/ - 单机8卡精度训练 ``` - # 单机8卡训练，XX.XX.XX.XX为当前IP - bash test/train.sh XX.XX.XX.XX + # 单机8卡训练 + bash test/train.sh ``` - 单机8卡的性能训练 ``` - # 单机8卡训练，XX.XX.XX.XX为当前IP - bash test/train_performance.sh XX.XX.XX.XX + # 单机8卡训练 + bash test/train_performance.sh ``` #### 训练结果 @@ -253,8 +249,8 @@ ${model-root-path}/pretrained/ #### 推理结果 | 芯片 | 卡数 | 性能-单步迭代耗时(s) | | ------------- | :--: |:-------------------: | -| 竞品A | 1p |11.2805 ± 0.0075| -| Atlas 800T A2 | 1p |11.0295 ± 1.3301| +| 竞品A | 1p |11.2805| +| Atlas 800T A2 | 1p |11.0295| # 变更说明 @@ -270,4 +266,6 @@ apt-get update && apt-get install -y gawk # CentOS/OpenEuler yum install -y gawk -``` \ No newline at end of file +``` + +2. 训练过程会自动下载inception权重，如果遇到网络问题等下载失败，可以本地下载后，手动将该权重文件放到日志指定路径 \ No newline at end of file diff --git a/model_examples/OpenDWM/test/train.sh b/model_examples/OpenDWM/test/train.sh index 9f38bab4..d1ba6cc2 100644 --- a/model_examples/OpenDWM/test/train.sh +++ b/model_examples/OpenDWM/test/train.sh @@ -35,8 +35,6 @@ python -m torch.distributed.run \ --nnodes $NNODES \ --nproc-per-node $NPROC_PER_NODE \ --node-rank $NODE_RANK \ - --master-addr $MASTER_ADDR \ - --master-port $MASTER_PORT \ src/dwm/train.py \ -c $CONFIG_FILE \ -o $OUTPUT_DIR 2>&1 | tee "${OUTPUT_DIR}/train.log" diff --git a/model_examples/OpenDWM/test/train_performance.sh b/model_examples/OpenDWM/test/train_performance.sh index bad73bbb..5221d69a 100644 --- a/model_examples/OpenDWM/test/train_performance.sh +++ b/model_examples/OpenDWM/test/train_performance.sh @@ -36,8 +36,6 @@ python -m torch.distributed.run \ --nnodes $NNODES \ --nproc-per-node $NPROC_PER_NODE \ --node-rank $NODE_RANK \ - --master-addr $MASTER_ADDR \ - --master-port $MASTER_PORT \ src/dwm/train.py \ -c $CONFIG_FILE \ -o $OUTPUT_DIR 2>&1 | tee "${OUTPUT_DIR}/train_performance.log" -- Gitee