-
Y. Xiong authored
* add torch.cuda.amp to fp16_utils and optimizers * use with context manager for autocast * add doc to explain the behavior differences between real amp and ours * fix docstring
fffc8757
* add torch.cuda.amp to fp16_utils and optimizers * use with context manager for autocast * add doc to explain the behavior differences between real amp and ours * fix docstring