While training a CNN network, this error raised
RuntimeError: "slow_conv2d_cpu" not implemented for 'ComplexDouble'. My datatype are
dtype=torch.complex128. So now, which
dtype do I need for my network to train? Is there any issue? Or to convert my data tensor in
dtype = torch.float16?
trainloader = create_data_loader(data_save, batch_size) ntotal_len = len(trainloader) for epoch in range(epochs): model.train() running_loss = 0.0 for target, labels in trainloader: target, labels = target.to(DEVICE), labels.to(DEVICE) outputs = model(target) loss = criterion(outputs, labels)