Allow for specify coda device in perplexity evaluation

[manuelbrack]

The current implementation of perplexity.compute does not allow to provide a specific coda device (e.g. cuda:1 instead of just cuda.

evaluate/metrics/perplexity/perplexity.py

Line 108 in 8dfe057

assert device in ["gpu", "cpu", "cuda"], "device should be either gpu or cpu."

Should probably be changed to something along the lines of:

assert device in ["gpu", "cpu", "cuda"] or re.search(r'^cuda:[0-9]+$', device), "device should be either gpu or cpu."