When your Encoder calls “x = self.enc_layers(…)”, you need to index the layer with [i], and you need to pass all of the parameters x, training, and mask. Not just ‘x’.
When your Encoder calls “x = self.enc_layers(…)”, you need to index the layer with [i], and you need to pass all of the parameters x, training, and mask. Not just ‘x’.