Skip to content

Commit 6105867

Browse files
authored
fix bifpn comments
1 parent ffca6c9 commit 6105867

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

efficientdet/model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)