Commit 0af3f56e authored by seyonechithrananda's avatar seyonechithrananda
Browse files

Update chemberta.py

parent b4e6646d

deepchem/models/torch_models/chemberta.py

+1 −1
Original line number Diff line number Diff line
@@ -42,7 +42,7 @@ class ChemBERTaforSequenceClassification(BertPreTrainedModel): 
    base_model_prefix = "roberta"



    def __init__(self, config, weight=None):

        super(ChemBERTa, self).__init__(config)

        super(ChemBERTaforSequenceClassification, self).__init__(config)

        self.num_labels = config.num_labels



        self.roberta = RobertaModel(config)

CRA Git | Maintained and supported by SUSTech CRA and CCSE