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 --- test/model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'test/model.py') diff --git a/test/model.py b/test/model.py index 5d60475..8122989 100644 --- a/test/model.py +++ b/test/model.py @@ -16,7 +16,7 @@ def test_default_signature(): 'runs', 'logs', 'RGB-GaitPart_80000_64_128_128_64_1_2_4_True_True_32_5_' '3_3_3_3_3_2_1_1_1_1_1_0_2_3_4_16_256_0.2_0.0001_0.9_' '0.999_0.001_500_0.9_CASIA-B_74_30_15_3_64_32_8_16') - assert model._signature == ('RGB-GaitPart_80000_0_64_128_128_64_1_2_4_True_' + assert model._checkpoint_sig == ('RGB-GaitPart_80000_0_64_128_128_64_1_2_4_True_' 'True_32_5_3_3_3_3_3_2_1_1_1_1_1_0_2_3_4_16_' '256_0.2_0.0001_0.9_0.999_0.001_500_0.9_CASIA-B' '_74_30_15_3_64_32_8_16') @@ -34,7 +34,7 @@ def test_default_signature_with_selector(): '3_3_3_3_3_2_1_1_1_1_1_0_2_3_4_16_256_0.2_0.0001_0.9_' '0.999_0.001_500_0.9_CASIA-B_74_30_15_3_64_32_bg-0\\d_' 'nm-0\\d_8_16') - assert model._signature == ('RGB-GaitPart_80000_0_64_128_128_64_1_2_4_True_' + assert model._checkpoint_sig == ('RGB-GaitPart_80000_0_64_128_128_64_1_2_4_True_' 'True_32_5_3_3_3_3_3_2_1_1_1_1_1_0_2_3_4_16_' '256_0.2_0.0001_0.9_0.999_0.001_500_0.9_CASIA-B' '_74_30_15_3_64_32_bg-0\\d_nm-0\\d_8_16') -- 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 --- test/model.py | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) (limited to 'test/model.py') diff --git a/test/model.py b/test/model.py index 8122989..f7fc57e 100644 --- a/test/model.py +++ b/test/model.py @@ -16,10 +16,10 @@ def test_default_signature(): 'runs', 'logs', 'RGB-GaitPart_80000_64_128_128_64_1_2_4_True_True_32_5_' '3_3_3_3_3_2_1_1_1_1_1_0_2_3_4_16_256_0.2_0.0001_0.9_' '0.999_0.001_500_0.9_CASIA-B_74_30_15_3_64_32_8_16') - assert model._checkpoint_sig == ('RGB-GaitPart_80000_0_64_128_128_64_1_2_4_True_' - 'True_32_5_3_3_3_3_3_2_1_1_1_1_1_0_2_3_4_16_' - '256_0.2_0.0001_0.9_0.999_0.001_500_0.9_CASIA-B' - '_74_30_15_3_64_32_8_16') + assert model._checkpoint_sig == ('RGB-GaitPart_0_80000_64_128_128_64_1_2_4_' + 'True_True_32_5_3_3_3_3_3_2_1_1_1_1_1_0_2_' + '3_4_16_256_0.2_0.0001_0.9_0.999_0.001_' + '500_0.9_CASIA-B_74_30_15_3_64_32_8_16') def test_default_signature_with_selector(): @@ -34,7 +34,8 @@ def test_default_signature_with_selector(): '3_3_3_3_3_2_1_1_1_1_1_0_2_3_4_16_256_0.2_0.0001_0.9_' '0.999_0.001_500_0.9_CASIA-B_74_30_15_3_64_32_bg-0\\d_' 'nm-0\\d_8_16') - assert model._checkpoint_sig == ('RGB-GaitPart_80000_0_64_128_128_64_1_2_4_True_' - 'True_32_5_3_3_3_3_3_2_1_1_1_1_1_0_2_3_4_16_' - '256_0.2_0.0001_0.9_0.999_0.001_500_0.9_CASIA-B' - '_74_30_15_3_64_32_bg-0\\d_nm-0\\d_8_16') + assert model._checkpoint_sig == ('RGB-GaitPart_0_80000_64_128_128_64_1_2_4_' + 'True_True_32_5_3_3_3_3_3_2_1_1_1_1_1_0_2_' + '3_4_16_256_0.2_0.0001_0.9_0.999_0.001_' + '500_0.9_CASIA-B_74_30_15_3_64_32_bg-0\\d_' + 'nm-0\\d_8_16') -- cgit v1.2.3