amnsbr / cubnm

A toolbox for biophysical network modeling on GPUs
https://cubnm.readthedocs.io
BSD 3-Clause "New" or "Revised" License
7 stars 3 forks source link

[ENH] Optimizer batch mode #10

Open amnsbr opened 2 months ago

amnsbr commented 2 months ago

Currently only one optimizer run can be executed at any given time. Sometimes (especially on GPUs) it would be more efficient if multiple optimizer runs are executed in parallel on the same GPU to use the full capacity of the device. These multiple runs can be different runs (with different seeds) of the same input data (i.e., same subject), or on different input data.