Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Enable access to training_step output for Callback.on_before_backward #20568

Open
zihaozou opened this issue Jan 30, 2025 · 0 comments
Open

Enable access to training_step output for Callback.on_before_backward #20568

zihaozou opened this issue Jan 30, 2025 · 0 comments
Labels
feature Is an improvement or enhancement needs triage Waiting to be triaged by maintainers

Comments

@zihaozou
Copy link

zihaozou commented Jan 30, 2025 •
edited by github-actions bot
Loading

Description & Motivation

It will be very useful if I can access the training_output in on_before_backward. For training gaussian splatting, I can implement the density control as a Callback and to retain_grad() for means2d

Pitch

No response

Alternatives

No response

Additional context

No response

cc @lantiga @Borda
@zihaozou zihaozou added feature Is an improvement or enhancement needs triage Waiting to be triaged by maintainers labels Jan 30, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
feature Is an improvement or enhancement needs triage Waiting to be triaged by maintainers
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@zihaozou