What is the intended way to run jobs on a GPU? #1
Description
Hi, thank you for your work.
It's unclear to me how one is meant to jobs on a GPU.
python job_script.py --task hop --device cuda --method maf_rgcn --outloc test_hopfield --nsims 1000 does not use the GPU.
Looking at the SNPE code (which this example seems to use), the default device is the cpu and the function sbi_training inside sbi4abm/inference/neural does not pass a device argument to the SNPE constructor.