Hi,
I tried the torch.cuda.amp.autocast in PyTorch and it works well for my model. Now I want to deploy my trained model in C++ with the nightly built libtorch (version 1.7.0). However, I cannot find a corresponding function for autocast in the libtorch library API 14.
May I ask what is the proper way to deploy a mixed precision model in libtorch?
Thanks,
Rui