@glaringlee: Can you please explain how you save a nn::Module in libtorch to file? And how to load that file in Python? When I try your suggestion and do
torch::save(model, "model.pt");

I get errors of the type
‘torch::serialize::OutputArchive’ is not derived from ‘std::basic_ostream<_CharT, _Traits>’
   archive << value;