ncnn
ncnn copied to clipboard
可以给生成的pnnx.py中的pytorch model 增加初始化权重默认入参么,方便迁移到其他地方使用
detail | 详细描述 | 詳細な説明
class Model(nn.Module): def init(self, pnnx_bin="pretrained/weights.bin"): super(Model, self).init() ... archive = zipfile.ZipFile(pnnx_bin, 'r') ...