diff options
| -rw-r--r-- | models/model.py | 2 | ||||
| -rw-r--r-- | utils/dataset.py | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/models/model.py b/models/model.py index 7cf6ed0..7e4edaf 100644 --- a/models/model.py +++ b/models/model.py @@ -312,7 +312,7 @@ class Model: self, dataset_config: Dict, dataloader_config: Dict, - ) -> Tuple[DataLoader, Dict[str: DataLoader]]: + ) -> Tuple[DataLoader, Dict[str, DataLoader]]: dataset_name = dataset_config.get('name', 'CASIA-B') if dataset_name == 'CASIA-B': gallery_dataset = self._parse_dataset_config( diff --git a/utils/dataset.py b/utils/dataset.py index e691157..ea678fb 100644 --- a/utils/dataset.py +++ b/utils/dataset.py @@ -75,7 +75,7 @@ class CASIAB(data.Dataset): self.views: np.ndarray[np.str_] # Labels, classes, conditions and views in dataset, # set of three attributes above - self.metadata = Dict[str, List[np.int64, str]] + self.metadata: Dict[str, List[np.int64, str]] # Dictionaries for indexing frames and frame names by clip name # and chip path when cache is on |