Update tutorials (#463)
--------- Co-authored-by: Dario Coscia <93731561+dario-coscia@users.noreply.github.com>
This commit is contained in:
committed by
FilippoOlivo
parent
8b797d589a
commit
bd9b49530a
@@ -479,10 +479,9 @@ class LabelTensor(torch.Tensor):
|
||||
|
||||
# Retrieve selected tensor and labels
|
||||
selected_tensor = super().__getitem__(index)
|
||||
if hasattr(self, "_labels"):
|
||||
original_labels=self._labels
|
||||
else:
|
||||
if not hasattr(self, "_labels"):
|
||||
return selected_tensor
|
||||
|
||||
original_labels = self._labels
|
||||
updated_labels = copy(original_labels)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user