Thanks. I wanted to make sure I wasn’t missing anything obvious. The pre-quantized model works because of the changes including
def __init__(self, *args, **kwargs):
    ...
    **self.add_relu = torch.nn.quantized.FloatFunctional()**
...
    def forward(self, x):
        identity = x
        out = self.conv1(x)
        ...
        **out = self.add_relu.add_relu(out, identity)**

        return out