File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -204,13 +204,13 @@ def _forward_fast_attention(self, inputs):
204204 # Weights for P5_0 and P6_1 to P5_1
205205 p5_w1 = self .p5_w1_relu (self .p5_w1 )
206206 weight = p5_w1 / (torch .sum (p5_w1 , dim = 0 ) + self .epsilon )
207- # Connections for P5_0 and P6_0 to P5_1 respectively
207+ # Connections for P5_0 and P6_1 to P5_1 respectively
208208 p5_up = self .conv5_up (self .swish (weight [0 ] * p5_in + weight [1 ] * self .p5_upsample (p6_up )))
209209
210210 # Weights for P4_0 and P5_1 to P4_1
211211 p4_w1 = self .p4_w1_relu (self .p4_w1 )
212212 weight = p4_w1 / (torch .sum (p4_w1 , dim = 0 ) + self .epsilon )
213- # Connections for P4_0 and P5_0 to P4_1 respectively
213+ # Connections for P4_0 and P5_1 to P4_1 respectively
214214 p4_up = self .conv4_up (self .swish (weight [0 ] * p4_in + weight [1 ] * self .p4_upsample (p5_up )))
215215
216216 # Weights for P3_0 and P4_1 to P3_2
You can’t perform that action at this time.
0 commit comments