mila-intel
/

protst-esm1b-for-sequential-classification

Feature Extraction

Model card Files Files and versions

Jiqing commited on Jan 26, 2024

Commit

d088998

·

verified ·

1 Parent(s): a58285f

Update modeling_protst.py

Files changed (1) hide show

modeling_protst.py +0 -1

modeling_protst.py CHANGED Viewed

@@ -162,7 +162,6 @@ class ProtSTForProteinPropertyPrediction(ProtSTPreTrainedModel):
         self.config = config
         self.protein_model = EsmForProteinRepresentation(config.protein_config)
-        self.logit_scale = nn.Parameter(torch.ones([]) * math.log(1 / 0.07))
         self.classifier = ProtSTHead(config.protein_config, out_dim=config.num_labels)
         self.post_init() # NOTE

         self.config = config
         self.protein_model = EsmForProteinRepresentation(config.protein_config)
         self.classifier = ProtSTHead(config.protein_config, out_dim=config.num_labels)
         self.post_init() # NOTE