search

steventrouble / EfficientZero

Fork of EfficientZero to use newer libraries and to fix a few runtime bugs. Also includes pretrained models!
GNU General Public License v3.0
7 stars 2 forks source link

FR: Work better on differently sized GPUs #4

Open steventrouble opened 2 years ago

steventrouble commented 2 years ago

Every GPU has different memory capabilities, but the current implementation doesn't account for this. We should compute (or take in as a flag) the amount of GPU memory per GPU, and automatically spin up the correct number of workers for the amount of GPU.