We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 936017c commit 3444865Copy full SHA for 3444865
advanced_source/cpp_frontend.rst
@@ -946,9 +946,9 @@ we use implement the `Adam <https://arxiv.org/pdf/1412.6980.pdf>`_ algorithm:
946
.. code-block:: cpp
947
948
torch::optim::Adam generator_optimizer(
949
- 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)));
950
torch::optim::Adam discriminator_optimizer(
951
- discriminator->parameters(), torch::optim::AdamOptions(5e-4).beta1(0.5));
+ discriminator->parameters(), torch::optim::AdamOptions(5e-4).betas(std::make_tuple(0.5, 0.5)));
952
953
.. note::
954
0 commit comments