diff options
Diffstat (limited to 'utils/sampler.py')
| -rw-r--r-- | utils/sampler.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/utils/sampler.py b/utils/sampler.py index 8dec846..0a177d1 100644 --- a/utils/sampler.py +++ b/utils/sampler.py @@ -1,5 +1,5 @@ import random -from typing import Iterator, Tuple +from collections.abc import Iterator import numpy as np from torch.utils import data @@ -11,7 +11,7 @@ class TripletSampler(data.Sampler): def __init__( self, data_source: CASIAB, - batch_size: Tuple[int, int] + batch_size: tuple[int, int] ): super().__init__(data_source) self.metadata_labels = data_source.metadata['labels'] |