search

NVlabs / LSGM

The Official PyTorch Implementation of "LSGM: Score-based Generative Modeling in Latent Space" (NeurIPS 2021)
Other
340 stars 49 forks source link

Number of GPUs not set correctly during sampling #3

Open jonasricker opened 2 years ago

jonasricker commented 2 years ago

Hi,

I believe that in this line args should be replaced by eval_args in order to compute num_gpus according to the passed arguments.

https://github.com/NVlabs/LSGM/blob/5eae2f385c014f2250c3130152b6be711f6a3a5a/evaluate_vada.py#L201

Thanks for sharing your work!