Hi,
using a 1D CVAE with a BCELoss function gives me the following (epoch, total number of batches, loss):
37 6912 tensor(318.8038, device='cuda:0', grad_fn=<NegBackward>)
38 6912 tensor(348.9748, device='cuda:0', grad_fn=<NegBackward>)
Traceback (most recent call last):
    data = data.cuda()

RuntimeError: CUDA error: device-side assert triggered

Does anyone know what is causing this?
The data loader statement is:
train_dataset2 = TensorDataset(X_train_2, y_train)
train_loader2 = DataLoader(train_dataset2, batch_size=BATCHSIZE, shuffle=True)

where X_train is the input 1d Vector and y_train is a linear array going from 1 to x where x is the length of the X_train array.
Chaslie