🔀 Merge branch 'main' into feat/sophia

token-tricksters · Aug 20, 2023 · 03b92fe · 03b92fe
2 parents 62bf578 + e437433
commit 03b92fe
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 2 deletions.
diff --git a/layers/AttentionLayer.py → AttentionLayer.py b/layers/AttentionLayer.py → AttentionLayer.py
diff --git a/classifier.py b/classifier.py
@@ -11,7 +11,7 @@
 from sklearn.metrics import classification_report, f1_score, recall_score, accuracy_score
 from torch.utils.tensorboard import SummaryWriter
 
-from layers.AttentionLayer import AttentionLayer
+from AttentionLayer import AttentionLayer
 # change it with respect to the original model
 from tokenizer import BertTokenizer
 from bert import BertModel

diff --git a/layers/__init__.py b/layers/__init__.py
diff --git a/multitask_classifier.py b/multitask_classifier.py
@@ -11,7 +11,7 @@
 from torch.utils.tensorboard import SummaryWriter
 
 from bert import BertModel
-from layers.AttentionLayer import AttentionLayer
+from AttentionLayer import AttentionLayer
 from optimizer import AdamW, SophiaH
 from tqdm import tqdm