ncnn icon indicating copy to clipboard operation
ncnn copied to clipboard

可以给生成的pnnx.py中的pytorch model 增加初始化权重默认入参么,方便迁移到其他地方使用

Open hmzjwhmzjw opened this issue 1 year ago • 0 comments

detail | 详细描述 | 詳細な説明

class Model(nn.Module): def init(self, pnnx_bin="pretrained/weights.bin"): super(Model, self).init() ... archive = zipfile.ZipFile(pnnx_bin, 'r') ...

hmzjwhmzjw avatar Apr 11 '24 06:04 hmzjwhmzjw