From 4d4557a32db774d47fbbb939f945ce78418f69dc Mon Sep 17 00:00:00 2001 From: weishengyu Date: Tue, 25 May 2021 23:38:59 +0800 Subject: [PATCH] dbg --- ppcls/arch/backbone/legendary_models/hrnet.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ppcls/arch/backbone/legendary_models/hrnet.py b/ppcls/arch/backbone/legendary_models/hrnet.py index d720fdfe..20cd01db 100644 --- a/ppcls/arch/backbone/legendary_models/hrnet.py +++ b/ppcls/arch/backbone/legendary_models/hrnet.py @@ -582,7 +582,8 @@ class HRNet(TheseusLayer): la1 = self.la1(conv2) - tr1 = self.tr1([la1]) + tr1 = self.tr1_1([la1]) + tr1 = self.tr1_2(tr1) st2 = self.st2(tr1) tr2 = self.tr2(st2) -- GitLab