def __init__(self, args, **kwargs):
    super(BasicMotionEncoder, self).__init__(**kwargs)
    cor_planes = args.corr_levels * (2 * args.corr_radius + 1)**2
    self.convc1 = create_update_Conv2d(c_in=cor_planes, c_out=256, k_size=1)
    self.convc2 = create_update_Conv2d(c_in=256, c_out=192, k_size=3)
    self.convf1 = create_update_Conv2d(c_in=2, c_out=128, k_size=7)
    self.convf2 = create_update_Conv2d(c_in=128, c_out=64, k_size=3)
    self.conv = create_update_Conv2d(c_in=64 + 192, c_out=128 - 2, k_size=3)