def __init__(self, args, **kwargs):
    super(SmallMotionEncoder, 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=96, k_size=1)
    self.convf1 = create_update_Conv2d(c_in=96, c_out=64, k_size=7)
    self.convf2 = create_update_Conv2d(c_in=64, c_out=32, k_size=3)
    self.conv = create_update_Conv2d(c_in=32, c_out=80, k_size=3)