def METHOD_NAME(self):
    """Initialize weights for cells."""
    for _, cell in self.cells_and_names():
        if isinstance(cell, nn.Conv2d):
            if cell is self.final_conv:
                cell.weight.set_data(init.initializer(init.Normal(), cell.weight.shape, cell.weight.dtype))
            else:
                cell.weight.set_data(init.initializer(init.HeUniform(), cell.weight.shape, cell.weight.dtype))
            if cell.bias is not None:
                cell.bias.set_data(init.initializer("zeros", cell.bias.shape, cell.bias.dtype))