From 3ee837ba86f01ba0d4e1c2ad382beff8bbe1e707 Mon Sep 17 00:00:00 2001 From: Federico Figari Tomenotti Date: Tue, 15 Apr 2025 15:56:02 +0200 Subject: [PATCH] Fix #1125 Corrected small typo on line 56 Fix of https://github.com/qubvel-org/segmentation_models.pytorch/issues/1125 --- docs/quickstart.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/quickstart.rst b/docs/quickstart.rst index 7fc04dd7..e6627b83 100644 --- a/docs/quickstart.rst +++ b/docs/quickstart.rst @@ -53,7 +53,7 @@ You are done! Now you can train your model with your favorite framework, or as s for images, gt_masks in dataloader: - predicted_mask = model(image) + predicted_mask = model(images) loss = loss_fn(predicted_mask, gt_masks) loss.backward()