Spaces:

d-matrix
/

dmx_perplexity

Sleeping

d-matrix commited on Feb 21

Commit

504216e

•

1 Parent(s): b0c3beb

Update dmx_perplexity.py

setting device for loss to avoid device mismatch when stacking

Files changed (1) hide show

dmx_perplexity.py CHANGED Viewed

@@ -118,7 +118,7 @@ class DmxPerplexity(evaluate.Metric):
                 else:
                     neg_log_likelihood = outputs.loss * trg_len
-            nlls.append(neg_log_likelihood)
             prev_end_loc = end_loc
             if end_loc == seq_len:

                 else:
                     neg_log_likelihood = outputs.loss * trg_len
+            nlls.append(neg_log_likelihood.to(device))
             prev_end_loc = end_loc
             if end_loc == seq_len: