Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.
[BugOnFeat] Resolve bug with Finetuning (#5744)
* resolve bug + add doc * Update pytorch_lightning/callbacks/finetuning.py * resolve bug * start adding more test * add more tests for finetuning callback functions * rename to flatten_modules * resolve doc * Update pytorch_lightning/callbacks/finetuning.py Co-authored-by: Jirka Borovec <[email protected]> * resolve comments * remove update on BoringModel * update on comments Co-authored-by: Carlos Mocholí <[email protected]> Co-authored-by: Jirka Borovec <[email protected]>
C
chaton committed
e425bf3ba93eb1654efd7cff9d3f6e54b40fe188
Parent: d05cdf8
Committed by GitHub <[email protected]>
on 2/4/2021, 6:36:54 PM