From 4befe59046fb3adf8ef8eb589999a74cf7136ff6 Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Wed, 6 Jan 2021 21:26:56 +0800 Subject: Add TensorBoard support --- test/model.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) create mode 100644 test/model.py (limited to 'test') diff --git a/test/model.py b/test/model.py new file mode 100644 index 0000000..472c210 --- /dev/null +++ b/test/model.py @@ -0,0 +1,21 @@ +import os + +from config import config +from models.model import Model + +conf = config + + +def test_default_signature(): + os.chdir('..') + model = Model(conf['system'], conf['model'], conf['hyperparameter']) + casiab = model._parse_dataset_config(conf['dataset']) + model._parse_dataloader_config(casiab, conf['dataloader']) + assert model.log_name == os.path.join( + '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_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_' + '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_CASIA-B_74_30_15_3_64_32_' + '8_16') -- cgit v1.2.3