A bug in dla.py

[changtaoli]

This is something wrong in the Tree class.

class Tree(nn.Module):
def init(self, block, in_channels, out_channels, level=1, stride=1):
super(Tree, self).init()
self.level = level
if level == 1:
self.root = Root(2*out_channels, out_channels)
self.left_node = block(in_channels, out_channels, stride=stride)
self.right_node = block(out_channels, out_channels, stride=1)
else:
self.root = Root((level+2)*out_channels, out_channels)
for i in reversed(range(1, level)):
subtree = Tree(block, in_channels, out_channels,
level=i, stride=stride)
self.setattr('level_%d' % i, subtree)
self.prev_root = block(in_channels, out_channels, stride=stride)
self.left_node = block(out_channels, out_channels, stride=1)
self.right_node = block(out_channels, out_channels, stride=1)

If a Tree is with level 3, then the output channels of the former subtree will not match the latter's in channels. What's more, the stride of the latter subtree will also be 2.