diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-01-07 18:37:43 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-01-07 18:37:43 +0800 |
| commit | 4a284084c253b9114fc02e1782962556ff113761 (patch) | |
| tree | d6ceff8da68b224186d84772ee6153353675bcfe /test | |
| parent | a27af5dfd58e7b48cf3bd063fa2b4b51ed1e0277 (diff) | |
Add typical training script and some bug fixes
1. Resolve deprecated scheduler stepping issue
2. Make losses in the same scale(replace mean with sum in separate triplet loss, enlarge pose similarity loss 10x)
3. Add ReLU when compute distance in triplet loss
4. Remove classes except Model from `models` package init
Diffstat (limited to 'test')
| -rw-r--r-- | test/cuda.py | 2 | ||||
| -rw-r--r-- | test/hpm.py | 2 | ||||
| -rw-r--r-- | test/rgb_part_net.py | 2 |
3 files changed, 3 insertions, 3 deletions
diff --git a/test/cuda.py b/test/cuda.py index ef0ea36..b1418c4 100644 --- a/test/cuda.py +++ b/test/cuda.py @@ -1,6 +1,6 @@ import torch -from models import RGBPartNet +from models.rgb_part_net import RGBPartNet P, K = 2, 4 N, T, C, H, W = P * K, 10, 3, 64, 32 diff --git a/test/hpm.py b/test/hpm.py index a68337d..0aefbb8 100644 --- a/test/hpm.py +++ b/test/hpm.py @@ -1,6 +1,6 @@ import torch -from models import HorizontalPyramidMatching +from models.hpm import HorizontalPyramidMatching T, N, C, H, W = 15, 4, 256, 32, 16 diff --git a/test/rgb_part_net.py b/test/rgb_part_net.py index 1d754a0..d0d4e91 100644 --- a/test/rgb_part_net.py +++ b/test/rgb_part_net.py @@ -1,6 +1,6 @@ import torch -from models import RGBPartNet +from models.rgb_part_net import RGBPartNet P, K = 2, 4 N, T, C, H, W = P * K, 10, 3, 64, 32 |