summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2021-01-12 14:26:03 +0800
committerJordan Gong <jordan.gong@protonmail.com>2021-01-12 14:26:03 +0800
commit61f79339fd0168bc200117daf0459764b93d0acb (patch)
tree484563cedd5cb65111bddd3042601b71af2a50aa
parent2b73aafe971f6014634eafbf130cc666531026bc (diff)
Some type hint fixes
-rw-r--r--models/model.py4
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