diff options
| -rw-r--r-- | models/model.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/models/model.py b/models/model.py index 617966f..af37bb2 100644 --- a/models/model.py +++ b/models/model.py @@ -182,11 +182,11 @@ class Model: def predict_all( self, iter_: int, - dataset_config: dict, + dataset_config: Dict, dataset_selectors: Dict[ str, Dict[str, Union[ClipClasses, ClipConditions, ClipViews]] ], - dataloader_config: dict, + dataloader_config: Dict, ) -> Dict[str, torch.Tensor]: self.is_train = False # Split gallery and probe dataset |