def __init__(self, hidden_dim=128, input_dim=192 + 128):
    super(SepConvGRU, self).__init__()
    self.convz1 = create_update_Conv2d(
        c_in=hidden_dim + input_dim, c_out=hidden_dim, k_size=(1, 5))
    self.convr1 = create_update_Conv2d(
        c_in=hidden_dim + input_dim, c_out=hidden_dim, k_size=(1, 5))
    self.convq1 = create_update_Conv2d(
        c_in=hidden_dim + input_dim, c_out=hidden_dim, k_size=(1, 5))

    self.convz2 = create_update_Conv2d(
        c_in=hidden_dim + input_dim, c_out=hidden_dim, k_size=(5, 1))
    self.convr2 = create_update_Conv2d(
        c_in=hidden_dim + input_dim, c_out=hidden_dim, k_size=(5, 1))
    self.convq2 = create_update_Conv2d(
        c_in=hidden_dim + input_dim, c_out=hidden_dim, k_size=(5, 1))