Finetuning with multiple GPUs

[francoisabcd]

Hi,

I have tried to fine tune the model using multiple GPUs, the code is not working.
It works if I only use one GPU.

Is it possible to finetune on multiple GPUs?

Thank you,

[shchur]

It might be possible to enable fine-tuning on multiple GPUs via extra_trainer_kwargs

chronos-forecasting/src/chronos/chronos2/pipeline.py

Line 109 in 67a6f91

**extra_trainer_kwargs,

But we have not tested this functionality and currently do not plan to actively work on multi-GPU support. Of course, feel free to share your experience and contributions are always welcome