Can you mention the constant values too?
BATCH_SIZE, VOCAB_SIZE, N_CLASSES
Unrelated to this question, some minor suggestions:

Use log_probs = model(x) instead of log_probs = model.forward(x) to make sure the necessary hooks are in place.
In pytorch 1.0, you do not need to have Variable and use the tensors directly.