search

IshmaelBelghazi / ALI

Adversarially Learned Inference
MIT License
311 stars 80 forks source link

Model can't be loaded without GPU #21

Open jcpeterson opened 6 years ago

jcpeterson commented 6 years ago

I need to load a pretrained model with CPU only, but it appears the cuda arrays are serialized as such. Is there any way around this?