ncnn torch.nn.utils.weight_norm breaks pnnx

7vhp5slm  于 3个月前  发布在  其他
关注(0)|答案(1)|浏览(42)

ref

torch.nn.utils.weight_norm

pytorch/pytorch#57289

qlzsbp2j

qlzsbp2j1#

THANKS TO NIHUI, this problem come from my model. it seems like changing the old version 'torch.nn.utils.weight_norm' to the new version 'torch.nn.utils.parametrizations.weight_norm' works, although i still can't fully convert my model to ncnn T_T

相关问题