wangjin2000 commited on
Commit
d82df2e
·
verified ·
1 Parent(s): f3b8a62

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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)