Add support to devices without CUDA

[samuelesabella]

No description provided.

[heyJatin]

Thanks a lot for these changes. It saved me a lot of time!

[belider]

@samuelesabella Thanks a lot for your contribution!
I tried to run your branch in google collab and faced the import error. I haven't been able to figure out how to fix this. Do you have any suggestions?

Traceback (most recent call last):
File "/content/Self-Correction-Human-Parsing/simple_extractor.py", line 24, in
import networks
File "/content/Self-Correction-Human-Parsing/networks/init.py", line 3, in
from .AugmentCE2P import resnet101
File "/content/Self-Correction-Human-Parsing/networks/AugmentCE2P.py", line 21, in
from ..modules import InPlaceABNSync
ImportError: attempted relative import beyond top-level package

[EgorOs]

Thank you very much @samuelesabella !
@belonel I managed to load checkpoint in Colab, here is an example https://colab.research.google.com/drive/1Rdm30i9ZuJEfTYMnFb-PRXWLkE6O_NVJ?usp=sharing