Closed Anikily closed 5 years ago
images, labels, edges, _ = batch labels = labels.long().cuda(non_blocking=True) edges = edges.long().cuda(non_blocking=True) preds = model(images)
images still be the torch.float before put into model.
images still be the torch.float before put into model.