文档
注册
评分
提单
论坛
小AI

__init__

功能说明

PrunneTorch类方法,对用户输入的模型进行类初始化。

函数原型

__init__(network, inputs)

参数说明

参数名

输入/返回值

含义

使用限制

network

输入

待剪枝模型实例。

必选。

数据类型:PyTorch模型。

inputs

输入

模型的输入数据,用于解析模型。

可选。

数据类型:Tensor。

调用示例

from msmodelslim.pytorch.prune.prune_torch import PruneTorch
model = torchvision.models.vgg16(pretrained=False)
model.eval()
prune_torch = PruneTorch(model, torch.ones([1, 3, 224, 224]).type(torch.float32))
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词