From 84748a01e6d3c26037412592ce147b7753ce6117 Mon Sep 17 00:00:00 2001 From: Francois Fleuret Date: Tue, 26 Jul 2022 12:47:26 +0200 Subject: [PATCH] Removed the Linear transformation since there is now w_o. --- mygpt.py | 1 - 1 file changed, 1 deletion(-) diff --git a/mygpt.py b/mygpt.py index 7f0c9e6..5370ffa 100755 --- a/mygpt.py +++ b/mygpt.py @@ -111,7 +111,6 @@ class MyGPT(nn.Module): nb_heads = nb_heads, causal = True, attention_dropout = dropout ), - nn.Linear(in_features = dim_model, out_features = dim_model), ), Residual( nn.LayerNorm(dim_model), -- 2.39.5