[solved] Assertion `srcIndex < srcSelectDimSize` failed on GPU for `torch.cat()`

I think the PyTorch error messages should be improved here. Having the wrong number of classes for nn.Embedding throws a bunch of C++ errors and returns CUDNN_STATUS_NOT_INITIALIZED on the latest version. Quite hard to debug this problem given these non-informative error messages.

3 Likes