Solved by tom in post #2 
              
              
                To train this model, you would likely need to convert it back to floating point and then train (because the quantized operators typically do not have autograd support). 
I am unaware of an automated way to do this. 
Best regards 
Thomas