Commit eb4a2c3a authored by liuzili97's avatar liuzili97 Committed by Kai Chen
Browse files

fix a bug in basicblock (#576)

* fix a bug in basicblock

* fix padding and dilation in basic block

* remove stride and dilation in conv2
parent 11e9c74c
......@@ -46,9 +46,7 @@ class BasicBlock(nn.Module):
planes,
planes,
3,
stride=stride,
padding=dilation,
dilation=dilation,
padding=1,
bias=False)
self.add_module(self.norm2_name, norm2)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment