From e07db719b97c6a44b01da78f65756a2cd6173223 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BC=A0=E4=B9=BE=E4=B9=BE?= <zhangqianqian46@huawei.com>
Date: Mon, 8 Sep 2025 09:47:27 +0800
Subject: [PATCH] fix_readme

---
 model_examples/OpenDWM/README.md                 | 10 ++++++----
 model_examples/OpenDWM/test/train.sh             |  2 +-
 model_examples/OpenDWM/test/train_performance.sh |  2 +-
 3 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/model_examples/OpenDWM/README.md b/model_examples/OpenDWM/README.md
index ce49c54a..a1ca7d66 100644
--- a/model_examples/OpenDWM/README.md
+++ b/model_examples/OpenDWM/README.md
@@ -129,10 +129,10 @@ OpenDWM是一种统一的多视角驾驶视频生成框架。通过融合单/多
     # 安装其他依赖项
     python -m pip install -r requirements.txt
 
-    # 安装tbe和hccl
+    # 安装tbe和hccl, 将 CANN 包目录记作 cann_root_dir
     pip uninstall te topi hccl -y
-    pip install $ASCEND_CUSTOM_PATH/latest/lib64/te-*-py3-none-any.whl
-    pip install $ASCEND_CUSTOM_PATH/latest/lib64/hccl-*-py3-none-any.whl
+    pip install {cann_root_dir}/latest/lib64/te-*-py3-none-any.whl
+    pip install {cann_root_dir}/latest/lib64/hccl-*-py3-none-any.whl
     ```
     
 6. 使用 patch 文件：
@@ -281,4 +281,6 @@ apt-get update && apt-get install -y gawk
 yum install -y gawk
 ```
 
-2. 训练过程会自动下载inception权重，如果遇到网络问题等下载失败，可以本地下载后，手动将该权重文件放到日志指定路径
\ No newline at end of file
+2. 训练过程会自动下载inception权重，如果遇到网络问题等下载失败，可以本地下载后，手动将该权重文件放到日志指定路径
+
+3. 我们支持的训练方式是单数据集nuScenes，若在执行`python -m pip install -r requirements.txt` 时由于网络原因下载kitti仓失败，注释掉对应依赖即可
\ No newline at end of file
diff --git a/model_examples/OpenDWM/test/train.sh b/model_examples/OpenDWM/test/train.sh
index d1ba6cc2..118d72e2 100644
--- a/model_examples/OpenDWM/test/train.sh
+++ b/model_examples/OpenDWM/test/train.sh
@@ -37,5 +37,5 @@ python -m torch.distributed.run \
     --node-rank $NODE_RANK \
     src/dwm/train.py \
     -c $CONFIG_FILE \
-    -o $OUTPUT_DIR 2>&1 | tee "${OUTPUT_DIR}/train.log"
+    -o $OUTPUT_DIR > "${OUTPUT_DIR}/train.log" 2>&1
     
\ No newline at end of file
diff --git a/model_examples/OpenDWM/test/train_performance.sh b/model_examples/OpenDWM/test/train_performance.sh
index 5221d69a..47c59de0 100644
--- a/model_examples/OpenDWM/test/train_performance.sh
+++ b/model_examples/OpenDWM/test/train_performance.sh
@@ -38,7 +38,7 @@ python -m torch.distributed.run \
     --node-rank $NODE_RANK \
     src/dwm/train.py \
     -c $CONFIG_FILE \
-    -o $OUTPUT_DIR 2>&1 | tee "${OUTPUT_DIR}/train_performance.log"
+    -o $OUTPUT_DIR > "${OUTPUT_DIR}/train_performance.log" 2>&1
 
 LOG_FILE="${OUTPUT_DIR}/train_performance.log"
 
-- 
Gitee