LakhNES icon indicating copy to clipboard operation
LakhNES copied to clipboard

Fix support for --cpu in generate.py

Open dartdart26 opened this issue 3 years ago • 0 comments

If the --cpu command line option is set, make sure we pass it propery to torch.load() so that if the system doesn't have CUDA, it uses the CPU. In order to do so, add the map_location parameter.

Additionally, remove the unused CUDA_MAJOR and CUDA_MINOR variables in proj_adaptive_softmax.py .

dartdart26 avatar Nov 02 '21 21:11 dartdart26