diff options
Diffstat (limited to 'utils/configuration.py')
| -rw-r--r-- | utils/configuration.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/utils/configuration.py b/utils/configuration.py index 445badd..e6bfaf2 100644 --- a/utils/configuration.py +++ b/utils/configuration.py @@ -8,12 +8,12 @@ from utils.dataset import ClipLabels, ClipConditions, ClipViews class SystemConfiguration(TypedDict): device: torch.device CUDA_VISIBLE_DEVICES: str - save_path: str + save_dir: str class DatasetConfiguration(TypedDict): name: str - path: str + root_dir: str train_size: int num_sampled_frames: int discard_threshold: int |