Thank you for the information.
Could you share the forward of  ConvGRUCell as I think that method fails according to the quote below.



 Zrufy:

Traceback (most recent call last):
  File "train.py", line 164, in <module>
    res, encoder_h_1, encoder_h_2, encoder_h_3)
  File "C:\Users\user\Anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\module.py", line 532, in __call__
    result = self.forward(*input, **kwargs)
  File "C:\Users\user\Desktop\ImageCompzByGRU-master\network.py", line 43, in forward
    hidden1 = self.rnn1(x, hidden1)