KevinGeertjens
/

bert-classification-model

Feature Extraction

bert-classification

Model card Files Files and versions Community

KevinGeertjens commited on Dec 1, 2022

Commit

dcb4bf1

•

1 Parent(s): 91c61bf

Update bert_classification_model.py

Files changed (1) hide show

bert_classification_model.py +0 -2

bert_classification_model.py CHANGED Viewed

@@ -4,8 +4,6 @@ from typing import Tuple
 from transformers import PreTrainedModel, BertModel
 from torch import nn
-from bert_classification_config import BertClassificationConfig
 class BertClassificationModel(PreTrainedModel):
     config_class = BertClassificationConfig

 from transformers import PreTrainedModel, BertModel
 from torch import nn
 class BertClassificationModel(PreTrainedModel):
     config_class = BertClassificationConfig