oh please specify them in list of lists, what are you fusing?
we support conv + bn, conv + bn + relu fusion.
For example if we have 0.conv, 0.bn, 0.relu and 1.conv, 1.bn we will call:
fuse_modules(model, [[“0.conv”, “0.bn”, “0.relu”], [“1.conv”, “1.bn”]])