diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-03-12 13:56:17 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-03-12 13:56:17 +0800 |
| commit | c74df416b00f837ba051f3947be92f76e7afbd88 (patch) | |
| tree | 02983df94008bbb427c2066c5f619e0ffdefe1c5 /test/part_net.py | |
| parent | 1b8d1614168ce6590c5e029c7f1007ac9b17048c (diff) | |
Code refactoring
1. Separate FCs and triplet losses for HPM and PartNet
2. Remove FC-equivalent 1x1 conv layers in HPM
3. Support adjustable learning rate schedulers
Diffstat (limited to 'test/part_net.py')
| -rw-r--r-- | test/part_net.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/test/part_net.py b/test/part_net.py index 25e92ae..fada2c4 100644 --- a/test/part_net.py +++ b/test/part_net.py @@ -64,7 +64,7 @@ def test_custom_part_net(): paddings=((2, 1), (1, 1), (1, 1), (1, 1)), halving=(1, 1, 3, 3), squeeze_ratio=8, - num_part=8) + num_parts=8) x = torch.rand(T, N, 1, H, W) x = pa(x) |