Solved by thyeros in post #4 
              
              
                I figured it out. For anyone with the same interest, hope this template helpful. Jerry, please comment if anything missing here. 
class qatCustom (nn.Module):
    def __init__(..., qconfig=None):
        super().__init__(....)
        self.qconifg = qconfig
        #skip this if there is no learnabl…