diff --git a/advanced_source/cpp_frontend.rst b/advanced_source/cpp_frontend.rst index 968afa01b2..f54c59afe4 100644 --- a/advanced_source/cpp_frontend.rst +++ b/advanced_source/cpp_frontend.rst @@ -1207,9 +1207,9 @@ optimizers are created, but before the training loop: .. code-block:: cpp torch::optim::Adam generator_optimizer( - generator->parameters(), torch::optim::AdamOptions(2e-4).beta1(0.5)); + generator->parameters(), torch::optim::AdamOptions(2e-4).betas(std::make_tuple(0.5, 0.5))); torch::optim::Adam discriminator_optimizer( - discriminator->parameters(), torch::optim::AdamOptions(2e-4).beta1(0.5)); + discriminator->parameters(), torch::optim::AdamOptions(2e-4).betas(std::make_tuple(0.5, 0.5))); if (kRestoreFromCheckpoint) { torch::load(generator, "generator-checkpoint.pt");