#Does unsloth provide "train_on_inputs"= False parameter similar to axolotl?

2 messages · Page 1 of 1 (latest)

fringe lichen
#

It changes the way the loss is calculated when training a lora -> i.e. it ignores the loss on input tokens and only trains the model on the completion token loss. Essentially allowing the model to concentrate entirely on learning to produce the output, at the cost of not learning to produce the input (which is what I want).

Reference: https://github.com/unslothai/unsloth/issues/735

GitHub

In axolotl, there's a config parameter you can set: train_on_inputs: false It changes the way the loss is calculated when training a lora -> i.e. it ignores the loss on input tokens and only...

vocal roost
#

Oh coincidence!