def __init__(self, hidden_dim=128, input_dim=192 + 128, **kwargs):
    super(ConvGRU, self).__init__(**kwargs)

    self.convz = create_update_Conv2d(
        c_in=hidden_dim + input_dim, c_out=hidden_dim, k_size=3)
    self.convr = create_update_Conv2d(
        c_in=hidden_dim + input_dim, c_out=hidden_dim, k_size=3)
    self.convq = create_update_Conv2d(
        c_in=hidden_dim + input_dim, c_out=hidden_dim, k_size=3)