Modified decoder and decoder for sequential architecture
This commit is contained in:
@@ -62,7 +62,7 @@ class Encoder(
|
||||
# 8) Layer Normalization
|
||||
x = self.__layer_norm_2(x)
|
||||
|
||||
return x
|
||||
return x,padding_mask
|
||||
|
||||
|
||||
# use eval to disable dropout ecc
|
||||
|
||||
Reference in New Issue
Block a user