Skip to content

Add a util function for training loop#34

Draft
SarahAlidoost wants to merge 3 commits intomainfrom
fix_33
Draft

Add a util function for training loop#34
SarahAlidoost wants to merge 3 commits intomainfrom
fix_33

Conversation

@SarahAlidoost
Copy link
Member

@SarahAlidoost SarahAlidoost commented Mar 18, 2026

closes #33

relates #13 using torch.utils.tensorboard

@SarahAlidoost SarahAlidoost changed the title Fix 33 Add a util function for training loop Mar 18, 2026
@SarahAlidoost
Copy link
Member Author

SarahAlidoost commented Mar 18, 2026

@rogerkuou I am implementing the training loop as a util function in this PR, also saving model and its config. I have to change the example notebook, but lets see how much we can change in other PR #27 and merge that one first.

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.

Add a util function for training loop

1 participant