diff --git a/train.py b/train.py index c6c377b..668914a 100644 --- a/train.py +++ b/train.py @@ -278,7 +278,7 @@ class GPT(nn.Module): x = block(x, ve, cos_sin, self.window_sizes[i]) x = norm(x) - softcap = 17 + softcap = 15 logits = self.lm_head(x) logits = logits.float() logits = softcap * torch.tanh(logits / softcap)