summaryrefslogtreecommitdiff
path: root/config.py
diff options
context:
space:
mode:
Diffstat (limited to 'config.py')
-rw-r--r--config.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/config.py b/config.py
index 0d98608..316ec50 100644
--- a/config.py
+++ b/config.py
@@ -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)