Update app.py
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_checkpoint1)
|
|
16 |
class MyModel(nn.Module):
|
17 |
def __init__(self):
|
18 |
super().__init__()
|
19 |
-
self.bert1 = EsmForSequenceClassification.from_pretrained(model_checkpoint1, num_labels=
|
20 |
# for param in self.bert1.parameters():
|
21 |
# param.requires_grad = False
|
22 |
self.bn1 = nn.BatchNorm1d(256)
|
|
|
16 |
class MyModel(nn.Module):
|
17 |
def __init__(self):
|
18 |
super().__init__()
|
19 |
+
self.bert1 = EsmForSequenceClassification.from_pretrained(model_checkpoint1, num_labels=3000)#3000
|
20 |
# for param in self.bert1.parameters():
|
21 |
# param.requires_grad = False
|
22 |
self.bn1 = nn.BatchNorm1d(256)
|