Make both torch.amp
and apex.amp
available as backend for mixed precision training#91
Merged
davidtvs merged 4 commits intodavidtvs:masterfrom NaleRaphael:dev/integrate-torch-ampJan 26, 2024
+369-42