Skip to content

Latest commit

 

History

14 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 
 
 

Repository files navigation

DJL_Pytorch_Demo

Simple demo showing pytorch model used in Amazon DJL

It comprises of 3 steps

  • Create a simple Linear Regression model using PyTorch
  • Serialize the model using TorchScript
  • Read that TorchScript model using AWS Deep Java Library for inference

About

Simple demo showing pytorch model used in Amazon DJL

Resources

Stars

4 stars

Watchers

2 watching

Forks

Releases

Packages

Contributors

Languages