Skip to content
New issue

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

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

Already on GitHub? # to your account

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

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
# for free to join this conversation on GitHub. Already have an account? # to comment
Labels
feature Is an improvement or enhancement needs triage Waiting to be triaged by maintainers
Projects
None yet
Development

No branches or pull requests

1 participant