kekpirat:

for layer in self.backbone_chunk1:
            x = layer(x)



looking at the code most likely it’s here:
x = self.quant0(x)
for layer in self.backbone_chunk1:
            x = layer(x)

only the first x is quantized in this case, instead you should have a quant x for each activation in the loop:
x = self.quant0(x)
for i, layer in enumerate(self.backbone_chunk1):
            x = layer(x)
            x = self.quants[i](x)

and define a list of quantstub instances with same length of self.backbone_chunk1 in init