Hello @gai,
thank you for the hint!
Now I have the full error message:
THCudaCheck FAIL file=/opt/conda/conda-bld/pytorch_1525909934016/work/aten/src/THC/THCTensorCopy.cu line=204 error=59 : device-side assert triggered
/opt/conda/conda-bld/pytorch_1525909934016/work/aten/src/THC/THCTensorIndex.cu:360: void indexSelectLargeIndex(TensorInfo<T, IndexType>, TensorInfo<T, IndexType>, TensorInfo<long, IndexType>, int, int, IndexType, IndexType, long) [with T = float, IndexType = unsigned int, DstDim = 1, SrcDim = 1, IdxDim = -2, IndexIsMajor = true]: block: [0,0,0], thread: [46,0,0] Assertion `srcIndex < srcSelectDimSize` failed.
This seems very similar to the error in this post Structured Learner which was not solved yet.
From the error description it looks like a dimension mismatch. I guess I have to check my df setup again.
I will keep digging deeper! 