diff --git a/model/model.py b/model/model.py index 398e93c..5c65ab7 100644 --- a/model/model.py +++ b/model/model.py @@ -98,7 +98,8 @@ def _transformerblocks(self, inputs, head_size, num_heads, Returns: A model layer. """ - inputs = Masking(mask_value=pad_value)(inputs) + #pad_value = -1 + #inputs = Masking(mask_value=pad_value)(inputs) x = MultiHeadAttention( key_dim=head_size, num_heads=num_heads, dropout=dropout)(inputs, inputs)