pytorch-lightning
Allow `training_step` in manual optimization to return general mappings
#21011
Merged
Merged
Allow `training_step` in manual optimization to return general mappings #21011
deependujha
merged
12 commits into
Lightning-AI:master
from
SkafteNicki:feature/manual_optimization_tensordict
change dict to mapping
41a4a67b
add a bit of testing
42418f87
justusschock
approved these changes
on 2025-07-31
Merge branch 'master' into feature/manual_optimization_tensordict
ef6b20ce
changelog
78e55098
use custom mapping
2e0def1c
![pre-commit-ci[bot]](https://avatars.githubusercontent.com/in/68672?v=4&s=60)
[pre-commit.ci] auto fixes from pre-commit.com hooks
ca515af7
bhimrazy
commented
on 2025-08-05
Update tests/tests_pytorch/trainer/optimization/test_manual_optimizat…
f476a3fb
Merge branch 'master' into feature/manual_optimization_tensordict
cdc42334
[pre-commit.ci] auto fixes from pre-commit.com hooks
ea1748a7
update
6e7be893
Merge branch 'master' into feature/manual_optimization_tensordict
b15733e8
Merge branch 'master' into feature/manual_optimization_tensordict
ed074ea5
Reviewers
Assignees
No one assigned
Labels
pl
Milestone
No milestone
Login to write a write a comment.
Login via GitHub