Save checkpoint with minimal overhead #11928
Unanswered
standard-aaron
asked this question in
Lightning Trainer API: Trainer, LightningModule, LightningDataModule
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
Uh oh!
There was an error while loading. Please reload this page.
-
I want to save a checkpoint of my model with absolutely no training whatsoever (i.e. just with randomly initialized weights) AFAICT there's no way to do this without invoking a Trainer method that requires arguments/setup beyond the model itself:
e.g.
-
.predict
that requires setting up a DataLoader, Dataset, etc.-
.fit
also requires a DataLoader (either as an argument or as a method to the mdoel)Is there a simpler way to save a checkpoint that won't require mocking up any data?
Beta Was this translation helpful? Give feedback.
All reactions