Hey,
I’m trying to use libtorch to create and save models. Following the DCGAN tutorial and this I create a basic script as given here
class TestImpl: torch::nn::Module
{
public:
  TestImpl(int inPlanes, int outClasses)
    : m_score(torch::nn::Conv2dOptions(inPlanes, outClasses, 1))
  {
    m_score->to(at::kCUDA);
    register_module("conv",m_score);
  }
  double forward(at::Tensor& input, at::Tensor& output)
  {
    cudaDeviceSynchronize();
    auto start = std::chrono::system_clock::now();

    output = m_score->forward(input);

    cudaDeviceSynchronize();
    auto end = std::chrono::system_clock::now();
    std::chrono::duration<double> elapsed_seconds = end-start;
    return double(elapsed_seconds.count());
  }
private:
  torch::nn::Conv2d m_score{nullptr};
};
TORCH_MODULE(Test);

int main(int argc, char** argv )
{
  Test net(3,10);
  int loop = 100;
  int skip = 20;
  float timing;
  at::Tensor outputs;
  at::Tensor inputs = torch::ones({1, 3, 1024, 1024},torch::TensorOptions().device(torch::kCUDA));

  for (auto i = 0; i < loop+skip; i++)
  {
      auto tmp = net->forward(inputs, outputs);//_pass(inputs, outputs);

      if(i >= skip)
          timing += tmp;
  }
  std::cout<< "Average forward pass in: " << timing/loop <<std::endl;
  torch::save(net, "chkValues.pt");
}

When I comment out the last line (torch::save) it compiles fine but un-commenting this throws out a lot of errors starting with:
In file included from libtorch_stable/include/torch/csrc/api/include/torch/nn/module.h:3:0,
                 from libtorch_stable/include/torch/csrc/api/include/torch/nn/cloneable.h:3,
                 from libtorch_stable/include/torch/csrc/api/include/torch/nn.h:3,
                 from libtorch_stable/include/torch/csrc/api/include/torch/all.h:7,
                 fromlibtorch_stable/include/torch/csrc/api/include/torch/torch.h:3,
                 from timing.cpp:36:
libtorch_stable/include/torch/csrc/api/include/torch/nn/pimpl.h: In instantiation of ‘torch::serialize::OutputArchive& torch::nn::operator<<(torch::serialize::OutputArchive&, const torch::nn::ModuleHolder<ModuleType>&) [with ModuleType = TestImpl]’:
libtorch_stable/include/torch/csrc/api/include/torch/serialize.h:43:11:   required from ‘void torch::save(const Value&, SaveToArgs&& ...) [with Value = Test; SaveToArgs = {const char (&)[13]}]’
timing.cpp:127:34:   required from here
libtorch_stable/include/torch/csrc/api/include/torch/nn/pimpl.h:179:18: error: no match for ‘operator<<’ (operand types are ‘torch::serialize::OutputArchive’ and ‘const std::shared_ptr<TestImpl>’)
   return archive << module.ptr();


Could someone please help me understand why this is happening?
I’m using libtorch-stable 1.4 version with cuda 9.2.