LibTorch's save_to & load_from function - Model Archiving

[Andrew_Edmondson]

Not sure exactly your need but have you seen this for saving to disk:

(libtorch) How to save model in MNIST cpp example?

I think, the example was written prior to the stable release of libtorch. The way you would implement the torch::nn::Module now is as follows struct NetImpl : torch::nn::Module { // replaced Net by NetImpl NetImpl() // replaced Net by NetImpl : conv1(torch::nn::Conv2dOptions(1, 10, /*kernel_size=*/5)), conv2(torch::nn::Conv2dOptions(10, 20, /*kernel_size=*/5)), fc1(320, 50), fc2(50, 10) { register_module("conv1", conv1); …

and this for transferring parameters between modules without saving:

Can libtorch load model from memory?

This is super helpful! Thank you!