summaryrefslogtreecommitdiff
path: root/test/part_net.py
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2021-03-12 13:59:05 +0800
committerJordan Gong <jordan.gong@protonmail.com>2021-03-12 13:59:05 +0800
commitd63b267dd15388dd323d9b8672cdb9461b96c885 (patch)
tree5095fc80fb93b946e4cfdee88258ab4fd49a8275 /test/part_net.py
parent08911dcb80ecb769972c2d2659c8ad152bbeb447 (diff)
parentc74df416b00f837ba051f3947be92f76e7afbd88 (diff)
Merge branch 'master' into python3.8
# Conflicts: # models/hpm.py # models/rgb_part_net.py # utils/configuration.py # utils/triplet_loss.py
Diffstat (limited to 'test/part_net.py')
-rw-r--r--test/part_net.py2
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)