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.

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

Split forward pass and loss calculation during training #5179

Open
mwindsp opened this issue Dec 11, 2023 · 1 comment
Open

Split forward pass and loss calculation during training #5179

mwindsp opened this issue Dec 11, 2023 · 1 comment
Labels
enhancement Improvements or good new features

Comments

@mwindsp
Copy link

mwindsp commented Dec 11, 2023

In the current implementation, the forward pass and loss calculation during training is done in one step. Both input and ground truth data are fed into the model.

In certain scenarios, it is however useful to split this into two distinct steps. E.g. I would like to calculate the loss twice using different ground truth data, which currently requires to perform the actual forward pass twice resulting in unnecessary computations.

Therefore, it would be beneficial, if the framework supports to split both steps.

@mwindsp mwindsp added the enhancement Improvements or good new features label Dec 11, 2023
@Programmer-RD-AI
Copy link
Contributor

Hi,
Is there anything supporting this that shows this will increase performance or provide any sort of benefit?
And could you elaborate on how this would work..?
Thank you

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement Improvements or good new features
Projects
None yet
Development

No branches or pull requests

2 participants