update model saving and loading to use torch jit

redhat-et / ray-on-the-edge-demo

A proof of concept for machine learning workflows using the Ray stack in the data center and at the edge.

Apache License 2.0

0 stars 2 forks source link

Closed MichaelClifford closed 2 years ago

MichaelClifford commented 2 years ago

Changed the way we load and save our models for training and inference .

review-notebook-app[bot] commented 2 years ago

Check out this pull request on

See visual diffs & provide feedback on Jupyter Notebooks.