🐛 moved attention layer to top level to include it in the submission …

…file
token-tricksters · Aug 16, 2023 · c014522 · c014522
1 parent 3cff0ff
commit c014522
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
@@ -10,7 +10,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
@@ -10,7 +10,7 @@
 from torch.utils.tensorboard import SummaryWriter
 
 from bert import BertModel
-from layers.AttentionLayer import AttentionLayer
+from AttentionLayer import AttentionLayer
 from optimizer import AdamW
 from tqdm import tqdm