d3ee410100
Add dedicated docs page for init-module ( #18416 )
2023-08-28 11:28:38 -04:00
dab373de54
Support loading a raw PyTorch state-dict checkpoint in Fabric ( #18049 )
...
Co-authored-by: Jirka Borovec <6035284+Borda@users.noreply.github.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-07-18 14:06:17 -04:00
c8656f1a27
docs: `fabric_methods.rst` using deprecated parameters ( #18015 )
2023-07-07 11:32:09 +02:00
1204006637
Document tensor shape requirements in `all_gather` ( #17816 )
2023-06-15 17:30:02 +02:00
24a3115995
Support empty weight initialization in `Fabric.init_module()` ( #17627 )
2023-06-07 18:33:53 +00:00
88cd100369
Address feedback for `Fabric.init_module()` (1/4) ( #17721 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-05-31 14:05:29 +00:00
54e8095a78
Split `init_module` into `init` + `sharded_model` ( #17488 )
2023-05-05 15:54:52 +02:00
4d17b5fe77
Improved model initialization API for Fabric ( #17462 )
...
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Jirka Borovec <6035284+Borda@users.noreply.github.com>
2023-04-26 11:25:33 -04:00
d48ec08d76
Update gradient clipping docs in Fabric ( #17470 )
2023-04-25 20:20:05 +00:00
20303e4736
Add Fabric.launch to Fabric methods section ( #17437 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-04-24 10:28:52 +02:00
54147e0745
Update Fabric docs navigation ( #16957 )
2023-03-06 16:13:51 +01:00
f2caa01bb3
Document gradient clipping in Fabric ( #16943 )
2023-03-05 17:03:57 +00:00
0e8ac7e1c9
docs: move fabric on its own ( #16742 )
...
* docs: move fabric to Lai
* update imports
* links
* drop link to Trainer
* own docs
* ci
* trigger
* prune cross-links
* cleaning
* cleaning
* template
* imports
* template
* path
* links
* tensorboardX
* plugins
* label
* drop fixme
* drop copy nb + examples
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Apply suggestions from code review
* try again
* rev
---------
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-03-01 12:36:14 +01:00
Adrian Wälchli