search

deepskies / DeepDiagnostics

Inference diagnostics for mostly SBI
MIT License
1 stars 0 forks source link

Unbind from CPU #53

Open voetberg opened 7 months ago

voetberg commented 7 months ago

Right now a lot of the code hard-codes sending arrays to cpu. This should be made dynamic and configurable with a "device" field in the config.