TypeError: LogisticRegression.fit() got an unexpected keyword argument 'gpu_per_trial'
- Dominant language
- Jupyter Notebook
- Stars
- 4.4k
- Forks
- 565
- Avg merge
- 5d 8m
- Merged PRs (30d)
- 17
Description
Hello,
I'm experiencing an issue when trying to run AutoFLAML with the `gpu_per_trial` argument. Here is the setup that I'm using:
```python
automl = AutoML()
automl_settings = {
'time_budget': 10,
'metric': 'accuracy',
'task': 'classification',
'estimator_list': 'auto',
'n_jobs': -1,
'use_ray': True,
'n_concurrent_trials': 10,
'gpu_per_trial': 0.1
}
automl.fit( X_train=X, y_train=y, **automl_settings)
```
When I run AutoFLAML with this setup, I get the following error:
`TypeError: LogisticRegression.fit() got an unexpected keyword argument 'gpu_per_trial'`
From my understanding, 'gpu_per_trial' is supposed to be a parameter for the RayTune backend to handle distribution of GPU resources across the trials. However, it seems like this argument is being passed to the fit method of the models AutoFLAML is training, which is causing the error.
I'm currently using AutoFLAML version 1.2.3. Is this a known issue, or is there something I'm missing in my configuration? Any guidance on this would be appreciated.
Thank you!
Best regards,
Andrej
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.