Skip to content
Snippets Groups Projects
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
No related branches found
No related tags found
No related merge requests found
...@@ -46,9 +46,7 @@ class BasicBlock(nn.Module): ...@@ -46,9 +46,7 @@ class BasicBlock(nn.Module):
planes, planes,
planes, planes,
3, 3,
stride=stride, padding=1,
padding=dilation,
dilation=dilation,
bias=False) bias=False)
self.add_module(self.norm2_name, norm2) self.add_module(self.norm2_name, norm2)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment