OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 00:21:07 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
a0207c8471
pytorch/torch/optim
History
Mauricio Villegas ece8267d2c Add back optim type hints that were lost when *.pyi files were removed (#136185) 
When stub files (`*.pyi`) were removed from `optim` (#125556, #125452), some types that existed are no longer available. This pull request adds them back.

Just for reference, these types are used in `pytorch-lightning`'s `LightningCLI`. Command line interfaces are created automatically, and having type hints make them nicer.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/136185
Approved by: https://github.com/janeyx99
2024-09-17 15:45:15 +00:00
..
_multi_tensor Optim package docstring fix (#129086) 2024-06-21 14:30:53 +00:00
__init__.py [BE][optim] Make pyright recognize exported symbols (#135043) 2024-09-04 21:53:46 +00:00
_adafactor.py Add Adafactor foreach impl (#132336) 2024-08-15 17:00:33 +00:00
_functional.py
adadelta.py Correct return type of grouping helper function in Optimizer (#133360) 2024-08-14 01:56:02 +00:00
adagrad.py move param's device check to _init_group for fused (#131153) 2024-08-17 04:49:47 +00:00
adam.py move param's device check to _init_group for fused (#131153) 2024-08-17 04:49:47 +00:00
adamax.py Correct return type of grouping helper function in Optimizer (#133360) 2024-08-14 01:56:02 +00:00
adamw.py move param's device check to _init_group for fused (#131153) 2024-08-17 04:49:47 +00:00
asgd.py Correct return type of grouping helper function in Optimizer (#133360) 2024-08-14 01:56:02 +00:00
lbfgs.py [BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771) 2024-08-01 17:07:14 +00:00
lr_scheduler.py Add back optim type hints that were lost when *.pyi files were removed (#136185) 2024-09-17 15:45:15 +00:00
nadam.py Correct return type of grouping helper function in Optimizer (#133360) 2024-08-14 01:56:02 +00:00
optimizer.py [BE] Clarify defaulting behavior in optimizer (#135384) 2024-09-06 21:52:55 +00:00
radam.py Correct return type of grouping helper function in Optimizer (#133360) 2024-08-14 01:56:02 +00:00
rmsprop.py Add back optim type hints that were lost when *.pyi files were removed (#136185) 2024-09-17 15:45:15 +00:00
rprop.py Correct return type of grouping helper function in Optimizer (#133360) 2024-08-14 01:56:02 +00:00
sgd.py Add back optim type hints that were lost when *.pyi files were removed (#136185) 2024-09-17 15:45:15 +00:00
sparse_adam.py [BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771) 2024-08-01 17:07:14 +00:00
swa_utils.py Make optim.swa.util content accessible from the torch.optim doc (#133393) 2024-08-21 00:43:46 +00:00