Do you think it’s because of the

(quant): QuantStub()
(dequant): DeQuantStub()
(module): QuantWrapper(
(quant): QuantStub()
(dequant): DeQuantStub()

Portion?
Anyways, now I fixed it, and found this error while using the Profiler

> Expected input images to be of floating type (in range [0, 1]), but found type torch.quint8 instead