add definition of scale for GPT

4 years ago · 641888f603
parent 37a722fe77
commit 641888f603
1 changed files with 1 additions and 0 deletions
--- a/model_zoo/official/nlp/gpt/src/gpt.py
+++ b/model_zoo/official/nlp/gpt/src/gpt.py
@ -361,6 +361,7 @@ class Block(nn.Cell):
    """
    def __init__(self, config, layer_idx):
        super(Block, self).__init__()
+        scale = 1.0
        self.layernorm1 = LayerNorm((config.embedding_size,)).to_float(config.compute_dtype)
        self.attention = Attention(config, scale, layer_idx)
        self.layernorm2 = LayerNorm((config.embedding_size,)).to_float(config.compute_dtype)