diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-03-12 20:30:25 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-03-12 20:34:00 +0800 |
| commit | 30b475c0a27e0f848743abf0f909607defc6a3ee (patch) | |
| tree | aaab163d3d76a835c32ce5014ce62637550d0b0d /test/part_net.py | |
| parent | 3d8fc322623ba61610fd206b9f52b406e85cae61 (diff) | |
| parent | e83ae0bcb5c763636fd522c2712a3c8aef558f3c (diff) | |
Merge branch 'data_parallel' into data_parallel_py3.8
# Conflicts:
# models/hpm.py
# models/model.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.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) |