Lightning-AI / pytorch-lightning UNCLAIMED

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

30975 0 0 Python

Add `configure_gradient_clipping` hook in `LightningModule` (#9584)

* init hook

* docs

* dep train args

* update tests

* doc

* doc

* .gitignore

* not dep

* add trainer args

* add & update tests

* fix tests

* pre-commit

* docs

* add docs

* add exception

* code review

* deepspeed

* update tests

* not

* try fix

* Apply suggestions from code review

* update deepspeed

* disable some tests

* disable some tests

* enable all tests

Rohit Gupta committed 4y ago

23e8b59ae7f586cc7e4eff3e73538ce878b991e6

Parent: 05b15e6

Committed by GitHub <[email protected]> on 10/13/2021, 2:45:13 PM