Langboat / Mengzi

Mengzi Pretrained Models
Apache License 2.0
534 stars 63 forks source link

padding部分不需要计算loss #56

Open Play2Boy opened 1 year ago

Play2Boy commented 1 year ago

Mengzi/examples/Mengzi_summary.ipynb

input_tensor["labels"] = target_tensor["input_ids"]

padding部分不需要计算loss