Hi @supriyar,
Yah, I did that replacement of quantstubs/dequantstubs with floatfunctional. I’ll drop it here
    > y = self.conv1x11(x)


    y = self.prelu1(y)
    y = self.norm1(y)
    w = self.depthwise_conv11(y)
    #w = w[:,:,:-2*2**1]
    w = w[:,:,:-4]
    y = self.depthwise_conv1(y)
    #y = y[:,:,:-2*2**1]
    y = y[:,:,:-4]
    y = self.prelu11(y)
    y = self.norm11(y)
    v = self.pointwise_conv1(y)
    #w = self.pointwise_conv_skp1(y)
    #z = self.dequant_tcnet(z)
    #w = self.dequant_tcnet(w)
    #v = self.dequant_tcnet(v)
    #x = self.dequant_tcnet(x)
    #z = z + w
    #x = x + v
    z = self.Qf_s.add(z,w)
    x = self.Qf_s.add(x,v)
    #x = self.quant_tcnet2(x)
    #z = self.quant(z)


as you can see now I removed stubs and using FloatFunctional but yet usage is not reduced
Thanks @supriyar,
Aravind