fix(contrib): call torch.cuda.is_available() in FP16_Optimizer guard - #2021
Open
andrewwhitecdw wants to merge 1 commit into
Open
fix(contrib): call torch.cuda.is_available() in FP16_Optimizer guard#2021andrewwhitecdw wants to merge 1 commit into
andrewwhitecdw wants to merge 1 commit into