Docs: enable nitpicky mode and fix broken links

docs/conf.py

@@ -46,12 +46,21 @@ def get_version():
 extensions = [
     "sphinx.ext.autodoc",
     "sphinx.ext.coverage",
+    "sphinx.ext.intersphinx",
     "sphinx.ext.napoleon",
     "sphinx.ext.viewcode",
     "sphinx.ext.mathjax",
     "autodocsumm",
 ]
 
+# Warn about all references where the target cannot be found.
+nitpicky = True
+nitpick_ignore = [
+    # Undocumented classes
+    ("py:class", "torch.FloatTensor"),
+    ("py:class", "torch.LongTensor"),
+]
+
 # Add any paths that contain templates here, relative to this directory.
 templates_path = ["_templates"]
 
@@ -109,6 +118,12 @@ def get_version():
 autoclass_content = "both"
 autodoc_typehints = "description"
 
+# sphinx.ext.intersphinx
+intersphinx_mapping = {
+    "python": ("https://docs.python.org/3", None),
+    "torch": ("https://docs.pytorch.org/docs/stable/", None),
+}
+
 # --- Work around to make autoclass signatures not (*args, **kwargs) ----------
 
 

segmentation_models_pytorch/losses/tversky.py

@@ -17,7 +17,7 @@ class TverskyLoss(DiceLoss):
     Args:
         mode: Metric mode {'binary', 'multiclass', 'multilabel'}
         classes: Optional list of classes that contribute in loss computation;
-        By default, all channels are included.
+            By default, all channels are included.
         log_loss: If True, loss computed as ``-log(tversky)`` otherwise ``1 - tversky``
         from_logits: If True assumes input is raw logits
         smooth:
@@ -28,8 +28,7 @@ class TverskyLoss(DiceLoss):
         gamma: Constant that squares the error function. Defaults to ``1.0``
 
     Return:
-        loss: torch.Tensor
-
+        torch.Tensor: loss
     """
 
     def __init__(