Don't fail with "AssertionError: Not enough CUDA devices to support n_devices 2" when there aren't 2 GPUs#4

JasonGross wants to merge 1 commit intoJoshEngels:mainJoshEngels/MultiDimensionalFeatures:mainfrom

JasonGross:patch-1JasonGross/MultiDimensionalFeatures:patch-1Copy head branch name to clipboard