fix srn fpn
parent
a5fbd902fe
commit
5f8c3e07d9
|
@ -104,7 +104,6 @@ class ResNet():
|
||||||
is_first=block == i == 0,
|
is_first=block == i == 0,
|
||||||
name=conv_name)
|
name=conv_name)
|
||||||
F.append(conv)
|
F.append(conv)
|
||||||
print("F:", F)
|
|
||||||
base = F[-1]
|
base = F[-1]
|
||||||
for i in [-2, -3]:
|
for i in [-2, -3]:
|
||||||
b, c, w, h = F[i].shape
|
b, c, w, h = F[i].shape
|
||||||
|
|
Loading…
Reference in New Issue