diff options
Diffstat (limited to 'utils/configuration.py')
| -rw-r--r-- | utils/configuration.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/utils/configuration.py b/utils/configuration.py index 84bd064..445badd 100644 --- a/utils/configuration.py +++ b/utils/configuration.py @@ -1,7 +1,9 @@ -from typing import TypedDict +from typing import TypedDict, Optional, Union import torch +from utils.dataset import ClipLabels, ClipConditions, ClipViews + class SystemConfiguration(TypedDict): device: torch.device @@ -15,6 +17,7 @@ class DatasetConfiguration(TypedDict): train_size: int num_sampled_frames: int discard_threshold: int + selector: Optional[dict[str, Union[ClipLabels, ClipConditions, ClipViews]]] num_input_channels: int frame_size: tuple[int, int] cache_on: bool |