Lightning-AI / pytorch-lightning UNCLAIMED

Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

0 0 152 Python

Direct support for compiled models (#15922)

* Direct support for compiled models

* Update test

* Update src/pytorch_lightning/core/module.py

Co-authored-by: Ethan Harris <ethanwharris@gmail.com>

Co-authored-by: Ethan Harris <ethanwharris@gmail.com>

Luca Antiga committed 3y ago

2992002bebd1a425f0c148a65ad98df2342b4ed5

Parent: e791749

Committed by GitHub <noreply@github.com> on 12/6/2022, 3:41:30 PM