Update models/peptide_classifiers.py
Browse files
models/peptide_classifiers.py
CHANGED
|
@@ -680,7 +680,7 @@ class HalfLifeModel:
|
|
| 680 |
|
| 681 |
|
| 682 |
def load_bindevaluator(checkpoint_path, device):
|
| 683 |
-
bindevaluator = BindEvaluator.load_from_checkpoint(checkpoint_path, n_layers=8, d_model=128, d_hidden=128, n_head=8, d_k=64, d_v=128, d_inner=64).to(device)
|
| 684 |
bindevaluator.eval()
|
| 685 |
for param in bindevaluator.parameters():
|
| 686 |
param.requires_grad = False
|
|
|
|
| 680 |
|
| 681 |
|
| 682 |
def load_bindevaluator(checkpoint_path, device):
|
| 683 |
+
bindevaluator = BindEvaluator.load_from_checkpoint(checkpoint_path, weights_only=False, n_layers=8, d_model=128, d_hidden=128, n_head=8, d_k=64, d_v=128, d_inner=64).to(device)
|
| 684 |
bindevaluator.eval()
|
| 685 |
for param in bindevaluator.parameters():
|
| 686 |
param.requires_grad = False
|