Modified-3D-UNet-Pytorch
Modified-3D-UNet-Pytorch copied to clipboard
unable to figure out training script
for epoch in range(max_epochs):
# Training
for orig, segmented in training_generator:
# Transfer to GPU
orig, segmented = local_batch.to(device), local_labels.to(device)
orig and segmented are torch tensors that represent the images that i need UNet to use to train. How do I implement the training script for this? I'm a bit confused.