test.py uses cpu if no gpu available

naoto0804 / pytorch-AdaIN

Unofficial pytorch implementation of 'Arbitrary Style Transfer in Real-time with Adaptive Instance Normalization' [Huang+, ICCV2017]

MIT License

1.08k stars 208 forks source link

Closed trvrm closed 6 years ago

trvrm commented 6 years ago

I had to make this 2 line change to get test.py to run even if there is no GPU available.

naoto0804 commented 6 years ago

Great, thanks!