[BUG] GPU training not working with non-default tabular presets #4196
Labels
bug
Something isn't working
module: tabular
Needs Triage
Issue requires Triage
priority: 1
High priority
Milestone
Bug Report Checklist
Describe the bug
Fitting the dataset using
num_gpus=1
only works whenpresets='medium_quality'
.This might have been introduced by the recent
ray
version bump #3774In the ray dashboard I see these events, and the tasks are stuck:
Warning: The following resource request cannot be scheduled right now: {'CPU': 12.0, 'GPU': 0.5}. This is likely due to all cluster resources being claimed by actors. Consider creating fewer actors or adding more nodes to this Ray cluster.
I only have 1 gpu, setting
num_gpus
to some value <1 seems to at least let the ray tasks run, albeit this doesn’t really work sincenum_gpus>=1
is expected downstreamFrom what I can tell this issue is present on google colab too, so it should be easily reproducible.
Expected behavior
GPU training should work even for more complex tabular model presets
To Reproduce
Run the tabular quickstart example notebook with
predictor = TabularPredictor(label=label).fit(train_data, num_gpus=1, presets='good_quality')
or better presetsInstalled Versions
autogluon==1.1.0
The text was updated successfully, but these errors were encountered: