diff --git a/PaddleSlim/docs/demo.md b/PaddleSlim/docs/demo.md index 0ea07eaf9c792687642bfd699ea0db9afafca871..c3c9001ed039c7f20d0a307ba3e948ff2c5f0e99 100644 --- a/PaddleSlim/docs/demo.md +++ b/PaddleSlim/docs/demo.md @@ -349,19 +349,21 @@ step1: 进入路径`PaddlePaddle/models/PaddleSlim/light_nas/`。 step2: (可选)按照[使用手册](https://github.com/PaddlePaddle/models/blob/develop/PaddleSlim/docs/usage.md#245-延时评估器生成方式)中说明的方法生成好延时评估器表格 `latency_lookup_table.txt`,放置到当前路径。 -step3: 在当前路径下,新建软链接指向上级目录的data: `ln -s ../data data`。 +step3: (可选)修改 `light_nas_space.py` 文件中的LATENCY_LOOKUP_TABLE_PATH, 更改为 LATENCY_LOOKUP_TABLE_PATH='latency_lookup_table.txt'。 -step4: 修改 `compress.yaml` 文件, 将参数 `server_ip` 设置为当前机器的 IP。 +step4: 在当前路径下,新建软链接指向上级目录的data: `ln -s ../data data`。 -step5: (可选)修改 `compress.yaml` 文件,将参数 `target_latency` 设置为用户的目标延时。 +step5: 修改 `compress.yaml` 文件, 将参数 `server_ip` 设置为当前机器的 IP。 -step6: 执行 `sh run.sh`, 可根据实际情况修改 `run.sh` 中的 `CUDA_VISIBLE_DEVICES`。 +step6: (可选)修改 `compress.yaml` 文件,将参数 `target_latency` 设置为用户的目标延时。 -step7: 修改 `light_nas_space.py` 文件中的 `LightNASSpace::init_tokens`, 使其返回step6中搜到的最优tokens。 +step7: 执行 `sh run.sh`, 可根据实际情况修改 `run.sh` 中的 `CUDA_VISIBLE_DEVICES`。 -step8: 修改 `compress.yaml` 文件,将 `compressor` 下的 `strategies` 去掉。 +step8: 修改 `light_nas_space.py` 文件中的 `LightNASSpace::init_tokens`, 使其返回step6中搜到的最优tokens。 -step9: 执行 `sh run.sh` 进行训练任务。 +step9: 修改 `compress.yaml` 文件,将 `compressor` 下的 `strategies` 去掉。 + +step10: 执行 `sh run.sh` 进行训练任务。 该示例基于 Flops 约束的两组结果如下: diff --git a/PaddleSlim/light_nas/light_nas_space.py b/PaddleSlim/light_nas/light_nas_space.py index f87541384eade0f1796ed7ac410deaf1e54ed2de..b99bb2ac5dd37a4c8f9a11db42d535d30deca58a 100644 --- a/PaddleSlim/light_nas/light_nas_space.py +++ b/PaddleSlim/light_nas/light_nas_space.py @@ -41,7 +41,7 @@ NAS_KERNEL_SIZE = [3, 5] NAS_FILTERS_MULTIPLIER = [3, 4, 5, 6] NAS_SHORTCUT = [0, 1] NAS_SE = [0, 1] -LATENCY_LOOKUP_TABLE_PATH = 'latency_lookup_table.txt' +LATENCY_LOOKUP_TABLE_PATH = None def get_bottleneck_params_list(var):