nas

模块

modelopt.torch.nas.algorithms

高级搜索和模型设计算法,帮助您优化模型。

modelopt.torch.nas.autonas

AutoNAS 模式的入口点。

modelopt.torch.nas.config

NAS模式的默认配置。

modelopt.torch.nas.conversion

模型剪枝的主要API和入口点。

modelopt.torch.nas.hparams

包含各种类型超参数实现的模块。

modelopt.torch.nas.mode

模块实现并描述了在NAS转换过程中可以使用的模式。

modelopt.torch.nas.modules

torch.nn模块的动态等效项。

modelopt.torch.nas.plugins

处理第三方模块的nas插件。

modelopt.torch.nas.registry

一个动态模块注册表,用于保存有关NAS可用DynamicModules的信息。

modelopt.torch.nas.search_space

DynamicSpace的扩展,用于处理可搜索单元及相关工具。

modelopt.torch.nas.traced_hp

模块用于指定具有跟踪功能的基本超参数。

modelopt.torch.nas.utils

用于修剪相关和搜索空间相关任务的实用函数。

NAS/基于剪枝的模型设计和优化的主模块。