import required torch and numpy libraries (#25483)

892f9ea0 · Erfan Zekri Esfahani · GitHub · fe3c8ab1 · 892f9ea0
Unverified Commit 892f9ea0 authored Aug 13, 2023 by Erfan Zekri Esfahani Committed by GitHub Aug 13, 2023
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

docs/source/en/tasks/semantic_segmentation.md docs/source/en/tasks/semantic_segmentation.md +4 -0

No files found.
--- a/docs/source/en/tasks/semantic_segmentation.md
+++ b/docs/source/en/tasks/semantic_segmentation.md
@@ -221,6 +221,10 @@ logits first, and then reshaped to match the size of the labels before you can c
 <pt>

 ```py
+>>> import numpy as np
+>>> import torch
+>>> from torch import nn
+
 >>> def compute_metrics(eval_pred):
 ...     with torch.no_grad():
 ...         logits, labels = eval_pred