Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -137,7 +137,9 @@ def compute_pseudo_perplexity(model, tokenizer, protein_seq, binder_seq):
|
|
| 137 |
with torch.no_grad():
|
| 138 |
outputs = model(masked_inputs, labels=labels)
|
| 139 |
loss = outputs.loss
|
| 140 |
-
|
|
|
|
|
|
|
| 141 |
# Loss is already averaged by the model
|
| 142 |
avg_loss = loss.item()
|
| 143 |
pseudo_perplexity = np.exp(avg_loss)
|
|
|
|
| 137 |
with torch.no_grad():
|
| 138 |
outputs = model(masked_inputs, labels=labels)
|
| 139 |
loss = outputs.loss
|
| 140 |
+
print("140: loss.shape:",loss.shape)
|
| 141 |
+
print("141: loss:",loss)
|
| 142 |
+
|
| 143 |
# Loss is already averaged by the model
|
| 144 |
avg_loss = loss.item()
|
| 145 |
pseudo_perplexity = np.exp(avg_loss)
|