diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-01-12 14:26:03 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-01-12 14:26:03 +0800 |
| commit | 61f79339fd0168bc200117daf0459764b93d0acb (patch) | |
| tree | 484563cedd5cb65111bddd3042601b71af2a50aa /models/model.py | |
| parent | 2b73aafe971f6014634eafbf130cc666531026bc (diff) | |
Some type hint fixes
Diffstat (limited to 'models/model.py')
| -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 |