Update modeling_protst.py
Browse files- modeling_protst.py +0 -1
modeling_protst.py
CHANGED
|
@@ -162,7 +162,6 @@ class ProtSTForProteinPropertyPrediction(ProtSTPreTrainedModel):
|
|
| 162 |
|
| 163 |
self.config = config
|
| 164 |
self.protein_model = EsmForProteinRepresentation(config.protein_config)
|
| 165 |
-
self.logit_scale = nn.Parameter(torch.ones([]) * math.log(1 / 0.07))
|
| 166 |
self.classifier = ProtSTHead(config.protein_config, out_dim=config.num_labels)
|
| 167 |
|
| 168 |
self.post_init() # NOTE
|
|
|
|
| 162 |
|
| 163 |
self.config = config
|
| 164 |
self.protein_model = EsmForProteinRepresentation(config.protein_config)
|
|
|
|
| 165 |
self.classifier = ProtSTHead(config.protein_config, out_dim=config.num_labels)
|
| 166 |
|
| 167 |
self.post_init() # NOTE
|