From c74df416b00f837ba051f3947be92f76e7afbd88 Mon Sep 17 00:00:00 2001
From: Jordan Gong <jordan.gong@protonmail.com>
Date: Fri, 12 Mar 2021 13:56:17 +0800
Subject: 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
---
 test/part_net.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'test/part_net.py')

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)
 
-- 
cgit v1.2.3