From 8ee391b65e2b48d777a268749f54b3aa9e4b9142 Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Wed, 13 Jan 2021 10:59:59 +0800 Subject: Add multiple checkpoints for different model and set default config value --- config.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) (limited to 'config.py') diff --git a/config.py b/config.py index 8a8d93a..7517097 100644 --- a/config.py +++ b/config.py @@ -13,7 +13,7 @@ config: Configuration = { 'dataset': { # Name of dataset (CASIA-B or FVG) 'name': 'CASIA-B', - # Path to dataset root + # Path to dataset root (required) 'root_dir': 'data/CASIA-B-MRCNN/SEG', # The number of subjects for training 'train_size': 74, @@ -88,9 +88,13 @@ config: Configuration = { 'model': { # Model name, used for naming checkpoint 'name': 'RGB-GaitPart', - # Restoration iteration from checkpoint - 'restore_iter': 0, - # Total iteration for training - 'total_iter': 80000, + # Restoration iteration from checkpoint (single model) + # 'restore_iter': 0, + # Total iteration for training (single model) + # 'total_iter': 80000, + # Restoration iteration (multiple models, e.g. nm, bg and cl) + 'restore_iters': (0, 0, 0), + # Total iteration for training (multiple models) + 'total_iter': (80_000, 80_000, 80_000), }, } -- cgit v1.2.3 From cdf2d2e00ff719e46b0d4bef46dabb1ddfe144dc Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Wed, 13 Jan 2021 11:15:04 +0800 Subject: Update config file and convert int to str when joining --- config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'config.py') diff --git a/config.py b/config.py index 7517097..791e888 100644 --- a/config.py +++ b/config.py @@ -95,6 +95,6 @@ config: Configuration = { # Restoration iteration (multiple models, e.g. nm, bg and cl) 'restore_iters': (0, 0, 0), # Total iteration for training (multiple models) - 'total_iter': (80_000, 80_000, 80_000), + 'total_iters': (80_000, 80_000, 80_000), }, } -- cgit v1.2.3