Skip to content

added a tutorial on custom models#102

Merged
avantikalal merged 1 commit intomainfrom
custom-model
Mar 10, 2025
Merged

added a tutorial on custom models#102
avantikalal merged 1 commit intomainfrom
custom-model

Conversation

@avantikalal
Copy link
Collaborator

@avantikalal avantikalal commented Feb 20, 2025

Added Tutorial 7 showing how to take an external pytorch model and wrap it in the LightningModel class to make it compatible with gReLU's downstream functions.

Some downstream functions require that the model head has an attribute n_tasks. To avoid the user forgetting to include this, I added a check in the BaseModel class.

Copy link
Collaborator

@suragnair suragnair left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks great!

@avantikalal avantikalal merged commit 1d61343 into main Mar 10, 2025
2 checks passed
@avantikalal avantikalal deleted the custom-model branch March 10, 2025 00:21
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants