Commit Graph

7 Commits

Author SHA1 Message Date
Vasilis Vryniotis 7e9e441843
Use TorchVision's Multi-weight Support and Model Registration API on Lightning ()
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
2022-09-09 20:04:57 +00:00
Rohit Gupta 8c6119fbce
Add auto wrapping support for `DDPFullyShardedStrategy` () 2022-09-05 19:07:26 +00:00
Adrian Wälchli 28e18881a9
Mark stage argument in hooks as required ()
Co-authored-by: rohitgr7 <rohitgr1998@gmail.com>
2022-09-01 15:47:40 +02:00
Rohit Gupta 6d00f31f0c
Add auto wrapping for `DDPFullyShardedNativeStrategy` () 2022-08-26 09:01:48 +00:00
Rohit Gupta 208512e6e9
Add docs for `fsdp_native` () 2022-08-12 09:14:07 +00:00
Rohit Gupta e21490b9bb
Update old PL links () 2022-06-21 16:38:04 +02:00
Jirka Borovec b58577fd4d
Future 3/n: docs adjustment ()
* docs: rename source >> source-PL

* docs: fix typing

* readthedocs

* update paths & codeowners

* source-pytorch

* ci

Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
2022-06-15 10:54:53 -04:00