|
|
@ -121,7 +121,7 @@ class BasicBlock(nn.Module):
|
|
|
|
super(BasicBlock, self).__init__()
|
|
|
|
super(BasicBlock, self).__init__()
|
|
|
|
|
|
|
|
|
|
|
|
assert cardinality == 1, 'BasicBlock only supports cardinality of 1'
|
|
|
|
assert cardinality == 1, 'BasicBlock only supports cardinality of 1'
|
|
|
|
assert base_width == 64, 'BasicBlock doest not support changing base width'
|
|
|
|
assert base_width == 64, 'BasicBlock does not support changing base width'
|
|
|
|
first_planes = planes // reduce_first
|
|
|
|
first_planes = planes // reduce_first
|
|
|
|
outplanes = planes * self.expansion
|
|
|
|
outplanes = planes * self.expansion
|
|
|
|
first_dilation = first_dilation or dilation
|
|
|
|
first_dilation = first_dilation or dilation
|
|
|
|