Skip to content

Commit

Permalink
dd
Browse files Browse the repository at this point in the history
  • Loading branch information
jfrery committed Dec 20, 2024
1 parent e75b376 commit b2a14fc
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions use_case_examples/lora_finetuning/eval.py
Original file line number Diff line number Diff line change
Expand Up @@ -125,12 +125,9 @@ def processed(example):
warmup_steps=10,
weight_decay=0.01,
prediction_loss_only=True,
logging_steps=50,
# evaluation_strategy is handled by LoraTrainer, not HuggingFace Trainer
report_to="none",
)


# Define a causal LM loss function
def causal_lm_loss(logits, labels, ignore_index=-100):
shift_logits = logits[..., :-1, :].contiguous()
Expand Down

0 comments on commit b2a14fc

Please sign in to comment.