pull/748/head
weishengyu 2021-05-25 23:44:11 +08:00
parent a2367a92b9
commit a0054caad4
1 changed files with 2 additions and 2 deletions

View File

@ -582,8 +582,8 @@ class HRNet(TheseusLayer):
la1 = self.la1(conv2)
tr1_1 = self.tr1_1([la1])
tr1_2 = self.tr1_2([la1])
tr1_1 = self.tr1_1(la1)
tr1_2 = self.tr1_2(la1)
st2 = self.st2([tr1_1, tr1_2])
tr2 = self.tr2(st2[-1])