Hello,
I’m kind of confused by the documentation. I am looking for a method to save / export my torch::nn::module (a simple CNN network of few layers) to a jit so I can later use torch::jit::load() instead of torch::load().
I’ve found torch::serialize::OutputArchive but I don’t understand how to use it and if its matches my problematic.
Do you have any insights on this ?
Thanks