Parameters: constraint_decoding (:obj:`bool`, `optional`, defaults to :obj:`False`): Whether to use Constraint Decoding structure_weight (:obj:`float`, `optional`, defaults to :obj:`None`):
| 69 | |
| 70 | @dataclass |
| 71 | class ConstraintSeq2SeqTrainingArguments(Seq2SeqTrainingArguments): |
| 72 | """ |
| 73 | Parameters: |
| 74 | constraint_decoding (:obj:`bool`, `optional`, defaults to :obj:`False`): |
| 75 | Whether to use Constraint Decoding |
| 76 | structure_weight (:obj:`float`, `optional`, defaults to :obj:`None`): |
| 77 | """ |
| 78 | constraint_decoding: bool = field(default=False, metadata={"help": "Whether to Constraint Decoding or not."}) |
| 79 | label_smoothing_sum: bool = field(default=False, |
| 80 | metadata={"help": "Whether to use sum token loss for label smoothing"}) |
| 81 | |
| 82 | |
| 83 | class ConstraintSeq2SeqTrainer(Seq2SeqTrainer): |
nothing calls this directly
no outgoing calls
no test coverage detected