search

amanchadha / iSeeBetter

iSeeBetter: Spatio-Temporal Video Super Resolution using Recurrent-Generative Back-Projection Networks | Python3 | PyTorch | GANs | CNNs | ResNets | RNNs | Published in Springer Journal of Computational Visual Media, September 2020, Tsinghua University Press
https://arxiv.org/abs/2006.11161
MIT License
359 stars 68 forks source link

Fix CPU mode #1

Closed rlaphoenix closed 4 years ago

rlaphoenix commented 4 years ago

torch.cuda.is_available() would return true if the user has a GPU with CUDA ready, argument check should be present here

amanchadha commented 4 years ago

You raise a very valid point. Merged your changes in.

Thanks!