diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2020-12-27 14:43:27 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2020-12-27 14:43:27 +0800 |
| commit | 3a8c48c722f687a47180221f807bd0a59c9f616e (patch) | |
| tree | 7f84f47b886b4a046612f3c7d05a48d5fef4902a /utils/dataset.py | |
| parent | 81b12f9357ca737599fbc11c732020666ad0873f (diff) | |
Add dataset selector to config type hint, change ClipLabels typo to ClipViews
Diffstat (limited to 'utils/dataset.py')
| -rw-r--r-- | utils/dataset.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/utils/dataset.py b/utils/dataset.py index 9f9229a..1a9d595 100644 --- a/utils/dataset.py +++ b/utils/dataset.py @@ -26,7 +26,7 @@ class CASIAB(data.Dataset): num_sampled_frames: int = 30, discard_threshold: int = 15, selector: Optional[dict[ - str, Union[ClipLabels, ClipConditions, ClipLabels] + str, Union[ClipLabels, ClipConditions, ClipViews] ]] = None, num_input_channels: int = 3, frame_size: tuple[int, int] = (64, 32), |