obuch with cross and graphic SAVING LORA

01a9d117 · Мазур Грета Евгеньевна · fd807d16 · 01a9d117 · 01a9d117
Commit 01a9d117 authored 1 week ago by Мазур Грета Евгеньевна
Hide whitespace changes
Inline Side-by-side

Showing

with 2 additions and 2 deletions
+2 -2
--- a/.ipynb_checkpoints/bertwithgoodgrad-checkpoint.py
+++ b/.ipynb_checkpoints/bertwithgoodgrad-checkpoint.py
@@ -43,7 +43,7 @@ class Classifiers(torch.nn.Module):
    def forward(self, hidden_state):
        cls_token = hidden_state[:, 0, :]  # Берем [CLS] токен
        return {
-            'safety': torch.softmax(self.safety(cls_token),
+            'safety': torch.softmax(self.safety(cls_token)),
            'attack': torch.softmax(self.attack(cls_token)) }
       


--- a/bertwithgoodgrad.py
+++ b/bertwithgoodgrad.py
@@ -43,7 +43,7 @@ class Classifiers(torch.nn.Module):
    def forward(self, hidden_state):
        cls_token = hidden_state[:, 0, :]  # Берем [CLS] токен
        return {
-            'safety': torch.softmax(self.safety(cls_token),
+            'safety': torch.softmax(self.safety(cls_token)),
            'attack': torch.softmax(self.attack(cls_token)) }