flash-attention icon indicating copy to clipboard operation
flash-attention copied to clipboard
verified publisher icon Dao-AILab

Update utils.py

Open adarshxs opened this issue 2 years ago • 1 comments

LightningLoggerBase -> Logger

LightningLoggerBase was deprecated https://github.com/Lightning-Universe/lightning-bolts/issues/962

adarshxs avatar Dec 08 '23 10:12 adarshxs

LightningLoggerBase is also used in train.py and eval.py https://github.com/Dao-AILab/flash-attention/blob/c9861a032d6a7d044eea33b09b62f9a5d3ae7266/training/src/train.py#L62 https://github.com/Dao-AILab/flash-attention/blob/c9861a032d6a7d044eea33b09b62f9a5d3ae7266/training/src/eval.py#L100

Pairshoe avatar Jan 10 '24 07:01 Pairshoe