Could you post the arguments of your model, i.e. input_dim, embedding_dim, and output_dim?
Also, could you post the exact PyTorch and CUDA versions (print(torch.__version__), print(torch.version.cuda))?