mirror of
https://github.com/wassname/peft.git
synced 2026-09-09 11:28:32 +08:00
Fix typo in examples/causal_language_modeling/peft_lora_clm_accelerate_ds_zero3_offload.py (#275) (#277)
Co-authored-by: rmilleti <rmilleti@amazon.com>
This commit is contained in:
co-authored by
rmilleti
parent
445940fb7b
commit
0422df466e
@@ -267,7 +267,7 @@ def main():
|
||||
tracemalloc.cpu_peaked + b2mb(tracemalloc.cpu_begin)
|
||||
)
|
||||
)
|
||||
train_epoch_loss = total_loss / len(eval_dataloader)
|
||||
train_epoch_loss = total_loss / len(train_dataloader)
|
||||
train_ppl = torch.exp(train_epoch_loss)
|
||||
accelerator.print(f"{epoch=}: {train_ppl=} {train_epoch_loss=}")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user