cpu code availability - Githubissues

yang-song / score_sde_pytorch

PyTorch implementation for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)

https://arxiv.org/abs/2011.13456

Apache License 2.0

1.58k stars 295 forks source link

cpu code availability #40

Open dgm2 opened 1 year ago

dgm2 commented 1 year ago

Hello Does this repo work on cpu? in particular the part in the OP folder? seems this only works only on Cuda? Are there some settings which can be set to enable running on CPU?

best regards