Hyun9898 commited on
Commit
fc7f78a
·
verified ·
1 Parent(s): 4127cd2

Delete modeling_bert_for_mid.py

Browse files
Files changed (1) hide show
  1. modeling_bert_for_mid.py +0 -12
modeling_bert_for_mid.py DELETED
@@ -1,12 +0,0 @@
1
-
2
- import torch
3
- import torch.nn as nn
4
-
5
- class BERT_for_MID(nn.Module):
6
- def __init__(self, bert, classifier):
7
- super(BERT_for_MID, self).__init__()
8
- self.bert = bert
9
- self.classifier = classifier
10
-
11
- def forward(self, input, attention_mask):
12
- return self.classifier(self.bert(input, attention_mask))