Skip to content
Discussion options

You must be logged in to vote
  1. .pt file in DJL referring to torchscript file. You can use python to trace the pytorch model and save it as .pt file
  2. .params is DJL parameter file, it's saved by DJL training code. You have to use DJL Block to load the .params file. Means, you need call Model.setBlock() first.
  3. The Model.load() method expect a folder or a torchscript file (for PyTorch). The file name extension is only being used to detect the model file when a folder is specified.

Replies: 1 comment

Comment options

You must be logged in to vote
0 replies
Answer selected by DialingSpoon527
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
2 participants