seq2seq-attn
seq2seq-attn copied to clipboard
Disable CUDA cross-GPU checks when 2 GPUs are used
Fixes the following error when training on two GPUs:
Assertion `THCTensor_(checkGPU)(state, 3, self_, src1, src2)'
Solution found here: https://github.com/torch/cutorch/issues/434#issuecomment-237281807