Fix optuna multigpu - Githubissues

Nixtla / neuralforecast

Scalable and user friendly neural :brain: forecasting algorithms.

https://nixtlaverse.nixtla.io/neuralforecast

Apache License 2.0

2.75k stars 316 forks source link

Closed jmoralez closed 5 months ago

jmoralez commented 5 months ago

Returns the fitted Trainer instead of using the model's attribute, which doesn't seem to be set when using multiple GPUs.

review-notebook-app[bot] commented 5 months ago

Check out this pull request on

See visual diffs & provide feedback on Jupyter Notebooks.