Merge branch 'master' of https://git.wmi.amu.edu.pl/s444465/projekt-glebokie

2023-02-12 19:36:00 +01:00 · 2023-02-12 19:36:00 +01:00 · d24d515841
commit d24d515841
parent f52c10479c b38c6db958
1 changed files with 3 additions and 0 deletions
--- a/gpt2.py
+++ b/gpt2.py
@ -87,6 +87,9 @@ class GPT2ClassificationHeadCustom(nn.Module):
        x = torch.relu(x)
        x = self.dropout(x)
        
+        x = torch.relu(x)
+        x = self.dropout(x)
+	
        x = self.out_proj(x)
        return x