NPU (华为昇腾)⚓︎
使用方法⚓︎
首先，请参考MMCV 安装带有 NPU 支持的 MMCV与 mmengine 。
使用如下命令，可以利用 8 个 NPU 训练模型（以 edsr 为例）：
bash tools/dist_train.sh configs/edsr/edsr_x2c64b16_1xb16-300k_div2k.py 8
或者，使用如下命令，在一个 NPU 上训练模型（以 edsr 为例）：
python tools/train.py configs/edsr/edsr_x2c64b16_1xb16-300k_div2k.py
经过验证的模型⚓︎
| Model | Dataset | PSNR | SSIM | Download |
|---|---|---|---|---|
| edsr_x2c64b16_1x16_300k_div2k | DIV2K | 35.83 | 0.94 | log |
注意:
- 如果没有特别标记，NPU 上的结果与使用 FP32 的 GPU 上的结果相同。
以上所有模型权重及训练日志均由华为昇腾团队提供
最后更新:
November 27, 2023
创建日期: November 27, 2023
创建日期: November 27, 2023