Upload torch saved model to wandb

[talhaanwarch]

I have trained a model and after training, i saved the model. now i want to upload the saved .pt file to wandb. How can i do that. I check this documentation https://pytorch-lightning.readthedocs.io/en/stable/api/pytorch_lightning.loggers.wandb.html but i cant find the option

[turian]

Just create a ModelCheckpoint and pass that as a callback to trainer.

[rohitgr7]

did you check the log_modelparameter of WandbLogger?
https://pytorch-lightning.readthedocs.io/en/stable/api/pytorch_lightning.loggers.wandb.html#pytorch_lightning.loggers.wandb.WandbLogger.params.log_model

[talhaanwarch]

i am saving my model like this way
torch.save(model.model.state_dict(), 'model.pt') but i dont know how to log this model. I dont want to log checkpoints

[talhaanwarch]

i think i got this, i can save it as wandb_logger.experiment.save('model.pt')