zeshengzong 25803d3a22 Optimize typing in lr_scheduler.py (#151219) ...

## Changes

- Add typing annotation in `lr_scheduler.py`

## Test Result

```bash
pytest test/optim/test_lrscheduler.py -vv
```

![image](https://github.com/user-attachments/assets/34a91965-ff3a-462a-9ab0-b46ad4b290e9)

Pull Request resolved: https://github.com/pytorch/pytorch/pull/151219
Approved by: https://github.com/janeyx99

2025-04-15 01:00:13 +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

Convert Tensor lr to 0-dim as needed for the optimizer to normally work (#145674)

2025-03-17 23:07:05 +00:00

_functional.py

PEP585 update - torch/nn torch/optim torch/package torch/profiler torch/serialization torch/sparse torch/xpu (#145175)

2025-01-21 16:57:27 +00:00

adadelta.py

Convert Tensor lr to 0-dim as needed for the optimizer to normally work (#145674)

2025-03-17 23:07:05 +00:00

adagrad.py

Convert Tensor lr to 0-dim as needed for the optimizer to normally work (#145674)

2025-03-17 23:07:05 +00:00

adam.py

[MPS] grad scaler (#150255)

2025-04-06 17:06:55 +00:00

adamax.py

Convert Tensor lr to 0-dim as needed for the optimizer to normally work (#145674)

2025-03-17 23:07:05 +00:00

adamw.py

PEP585 update - torch/nn torch/optim torch/package torch/profiler torch/serialization torch/sparse torch/xpu (#145175)

2025-01-21 16:57:27 +00:00

asgd.py

Convert Tensor lr to 0-dim as needed for the optimizer to normally work (#145674)

2025-03-17 23:07:05 +00:00

lbfgs.py

Convert Tensor lr to 0-dim as needed for the optimizer to normally work (#145674)

2025-03-17 23:07:05 +00:00

lr_scheduler.py

Optimize typing in lr_scheduler.py (#151219)

2025-04-15 01:00:13 +00:00

nadam.py

Convert Tensor lr to 0-dim as needed for the optimizer to normally work (#145674)

2025-03-17 23:07:05 +00:00

optimizer.py

Convert Tensor lr to 0-dim as needed for the optimizer to normally work (#145674)

2025-03-17 23:07:05 +00:00

radam.py

Convert Tensor lr to 0-dim as needed for the optimizer to normally work (#145674)

2025-03-17 23:07:05 +00:00

rmsprop.py

Convert Tensor lr to 0-dim as needed for the optimizer to normally work (#145674)

2025-03-17 23:07:05 +00:00

rprop.py

Convert Tensor lr to 0-dim as needed for the optimizer to normally work (#145674)

2025-03-17 23:07:05 +00:00

sgd.py

Convert Tensor lr to 0-dim as needed for the optimizer to normally work (#145674)

2025-03-17 23:07:05 +00:00

sparse_adam.py

Convert Tensor lr to 0-dim as needed for the optimizer to normally work (#145674)

2025-03-17 23:07:05 +00:00

swa_utils.py

Revert "Fix non-bitwise type annotations for Tensor operators (see #145838) (#146845)"

2025-02-18 19:01:27 +00:00