Add `gpu_id` option to command line arguments `MLExperiment`

mle-infrastructure / mle-toolbox

Lightweight Tool to Manage Distributed ML Experiments 🛠

https://mle-infrastructure.github.io/mle_toolbox/toolbox/

MIT License

3 stars 1 forks source link

Add `gpu_id` option to command line arguments `MLExperiment` #98

Open RobertTLange opened 2 years ago

RobertTLange commented 2 years ago

Set cuda visible devices in experiment startup.

RobertTLange commented 2 years ago

Additionally, make it possible to overwrite config file with option. E.g. --lrate 0.001 should set the lrate in train_config.

RobertTLange commented 2 years ago

More general possibility to add device_config to the base configuration file. If it is loaded by the MLExperiment startup, automatically setup the GPU visibility, etc.