We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 627fbdb commit d645dceCopy full SHA for d645dce
docs/quickstart.rst
@@ -53,7 +53,7 @@ You are done! Now you can train your model with your favorite framework, or as s
53
54
for images, gt_masks in dataloader:
55
56
- predicted_mask = model(image)
+ predicted_mask = model(images)
57
loss = loss_fn(predicted_mask, gt_masks)
58
59
loss.backward()
0 commit comments