cuda device side assert triggered pytorch

Question

i want to train 5 network at the same time

model = VAE_NET(device=device).to(device)
netG = UNet(in_channel = 3, out_channel = 3, n_kernel = 64).to(device)
p_d =