This is not a too big deal, but makes it hard to check the results, except if you have working cuda, which is not always the case.
Workaround:
Change line 256 in run.py to the following:
state = torch.load(load_path, map_location=torch.device('cpu'))
It would be nice to have the run.py to check itself whether cuda is available and set the parameter accordingly if needed.
This is not a too big deal, but makes it hard to check the results, except if you have working cuda, which is not always the case. Workaround: Change line 256 in run.py to the following:
It would be nice to have the run.py to check itself whether cuda is available and set the parameter accordingly if needed.