Solved by Zafar in post #2 
              
              
                Yes it is. I am not sure what exactly your model is, but here is an example on how you could make a model with two inputs in the forward 
class SomeAwesomeModel(nn.Module):
    def __init__(self):
        self.quant_x = torch.quantization.QuantStub()
        self.quant_y = torch.quantization.QuantSt…