diff --git a/paddleslim/nas/nas_api.md b/paddleslim/nas/nas_api.md index 46cf9d2d63a76e3825e7cb0ffaae06bb1569846a..4467c45d8bba3030a1089b4e762dad244368c8fa 100644 --- a/paddleslim/nas/nas_api.md +++ b/paddleslim/nas/nas_api.md @@ -3,11 +3,11 @@ ## SANAS API文档 ## class SANAS +SANAS(Simulated Annealing Neural Architecture Search)是基于模拟退火算法进行模型结构搜索的算法,一般用于离散搜索任务。 --- >paddleslim.nas.SANAS(configs, server_addr, init_temperature, reduce_rate, search_steps, save_checkpoint, load_checkpoint, is_server) -SANAS(Simulated Annealing Neural Architecture Search)是基于模拟退火算法进行模型结构搜索的算法,一般用于离散搜索任务。 **参数:** - **configs(list):** 搜索空间配置列表,格式是`[(key, {input_size, output_size, block_num, block_mask})]`或者`[(key)]`(MobileNetV2、MobilenetV1和ResNet的搜索空间使用和原本网络结构相同的搜索空间,所以仅需指定`key`即可), `input_size` 和`output_size`表示输入和输出的特征图的大小,`block_num`是指搜索网络中的block数量,`block_mask`是一组由0和1组成的列表,0代表不进行下采样的block,1代表下采样的block。 更多paddleslim提供的搜索空间配置可以参考。