How to use mixed precision training models in libtorch

C++
1

Thanks to anyone for providing clues

2

An ancient snippet from the author of AMP that might be relevant… autocast get OOM with c++ api · Issue #46649 · pytorch/pytorch · GitHub

3

thanks any way, after at::autocast::set_enabled(true);The model accuracy has decreased。