模块

模块

modelopt.torch.nas.modules.container

动态顺序实现(基于torch.nn.modules.container的可变深度)。

modelopt.torch.nas.modules.conv

基于 torch.nn.modules.conv 的动态卷积实现。

modelopt.torch.nas.modules.linear

基于torch.nn.modules.linear的动态线性实现。

modelopt.torch.nas.modules.norm

基于torch.nn.modules中的norm模块的动态norm实现。

modelopt.torch.nas.modules.utils

用于实用函数的内部模块。

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