Shortcuts

Function torch::jit::load(std::shared_ptr<caffe2::serialize::ReadAdapterInterface>, std::optional<c10::Device>, bool)

Function Documentation

Module torch::jit::load(std::shared_ptr<caffe2::serialize::ReadAdapterInterface> rai, std::optional<c10::Device> device = std::nullopt, bool load_debug_files = true)

Loads a serialized Module from the given shared_ptr rai.

The reader adapter, which is for customized input stream, must contain a serialized Module, exported either via ScriptModule.save() in Python or torch::jit::ExportModule in C++.

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources