Specify cuda device for model storage

eth-easl / modyn

Modyn is a research-platform for training ML models on growing datasets.

MIT License

22 stars 3 forks source link

Specify cuda device for model storage #477

Open MaxiBoether opened 1 month ago

MaxiBoether commented 1 month ago

https://github.com/eth-easl/modyn/blob/2b80acab4ee67f0ae9554a8a58845246deb0aba3/modyn/model_storage/internal/model_storage_manager.py#L177

I am not sure how this made it into the codebase, but we need to make this configurable. I wasted time trying to debug an invalid cuda device error here.