summaryrefslogtreecommitdiff
path: root/test/model.py
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2021-01-13 11:15:04 +0800
committerJordan Gong <jordan.gong@protonmail.com>2021-01-13 11:15:04 +0800
commitcdf2d2e00ff719e46b0d4bef46dabb1ddfe144dc (patch)
tree908f5549bf4497c5af1f76b1490adacdc8d2f379 /test/model.py
parent8ee391b65e2b48d777a268749f54b3aa9e4b9142 (diff)
Update config file and convert int to str when joining
Diffstat (limited to 'test/model.py')
-rw-r--r--test/model.py17
1 files changed, 9 insertions, 8 deletions
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')