There's no way to put, and keep a model on the CUDA device

C++
2

I think you forgot to call register_module in the constructor


Critic_Net() 
{
   lin1 = register_module("lin1", torch::nn::Linear(427, 42));
   lin2 = register_module("lin2", torch::nn::Linear(42, 286));
   lin3 = register_module("lin3", torch::nn::Linear(286, 1));
}
2 Likes