ckpt_path
in Trainer
accepts URIs to automatically load checkpoints from remote paths
#19849
Labels
ckpt_path
in Trainer
accepts URIs to automatically load checkpoints from remote paths
#19849
Description & Motivation
If I set up a Trainer with a
WandbLogger
and setlog_model=True
, I get that my model is saved locally and in a W&B server.If I want to retrieve the model from the server I have to use the W&B
use_artifact
methods and the download methods to first retrieve the model, as described hereWhat I would like to have instead is to not specify the W&B logic explicitly and download the model by just passing a URI with the remote string, (e.g.
wandb://user/project/model-run_id:version
) as theckpt_path
in theTrainer
, along with theWandbLogger
.In short, I would like to do something like that:
Pitch
Specify a URI to a remote resource, automatically download it, and load it in the
Trainer
.Alternatives
Write explicitly the W&B logic to download the model and then pass the checkpoint path to the
Trainer
.Additional context
No response
cc @Borda
The text was updated successfully, but these errors were encountered: