I want to change the datatype of the mxnet architecture

[Foreist]

I am using the mxnet-cu101mkl version. Learning is in progress with the gpu version. Inference must inevitably go with the cpu version. I want to change the model architecture trained with float32 to float16. How can I change the datatype of nn.Conv2d?

(Note that the cpu version cannot use cast.)