Skip to content

move loss out of the model#9

Merged
ayeganov merged 2 commits intomainfrom
feat/loss_separated
Sep 1, 2025
Merged

move loss out of the model#9
ayeganov merged 2 commits intomainfrom
feat/loss_separated

Conversation

@ayeganov
Copy link
Contributor

What Does This PR Do?

In this PR I move out the loss calculation from the model's forward pass. The inference no longer requires loss as well.

@ayeganov ayeganov self-assigned this Aug 29, 2025
@ayeganov ayeganov requested a review from dariocazzani August 29, 2025 20:14
@dariocazzani dariocazzani self-requested a review September 1, 2025 08:38
@ayeganov ayeganov merged commit e9a8d3d into main Sep 1, 2025
2 checks passed
@ayeganov ayeganov deleted the feat/loss_separated branch September 1, 2025 17:15
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants