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:
Robert Milletich
2023-04-07 17:49:35 +02:00
committed by GitHub
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=}")