Skip to content

fix(contrib): call torch.cuda.is_available() in FP16_Optimizer guard - #2021

Open
andrewwhitecdw wants to merge 1 commit into
NVIDIA:masterfrom
andrewwhitecdw:sweep/bugfix-cuda-availability-check
Open

fix(contrib): call torch.cuda.is_available() in FP16_Optimizer guard#2021
andrewwhitecdw wants to merge 1 commit into
NVIDIA:masterfrom
andrewwhitecdw:sweep/bugfix-cuda-availability-check

fix(contrib): call torch.cuda.is_available() in FP16_Optimizer guard

40c0f0e
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs