LakhNES
LakhNES copied to clipboard
Fix support for --cpu in generate.py
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 .