diff options
| -rw-r--r-- | config.py | 4 | ||||
| -rw-r--r-- | utils/configuration.py | 4 |
2 files changed, 4 insertions, 4 deletions
@@ -9,14 +9,14 @@ config: Configuration = { # GPU(s) used in training or testing, if CUDA enabled 'CUDA_VISIBLE_DEVICES': '0', # Directory used in training or testing for temporary storage - 'save_path': 'runs', + 'save_dir': 'runs', }, # Dataset settings 'dataset': { # Name of dataset (CASIA-B or FVG) 'name': 'CASIA-B', # Path to dataset root - 'path': 'dataset/output/CASIA-B', + 'root_dir': 'dataset/output/CASIA-B', # The number of subjects for training 'train_size': 74, # Number of sampled frames per sequence (Training only) 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 |