feat: enable GPU device selection

rickstaa / stable-learning-control

A framework for training theoretically stable (and robust) Reinforcement Learning control algorithms.

https://rickstaa.dev/stable-learning-control

MIT License

6 stars 1 forks source link

feat: enable GPU device selection #406

Closed rickstaa closed 7 months ago

rickstaa commented 7 months ago

This pull request enhances the functionality by allowing users to specify the GPU device they wish to use for training. This feature supports multiple GPUs, enabling more flexible and efficient resource utilization.