Skip to content

Commit

Permalink
'pdqn_agent_x(#1)'
Browse files Browse the repository at this point in the history
  • Loading branch information
baijinqiu committed Dec 24, 2023
1 parent 5b14523 commit c17f588
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions docs/source/documents/api/learners/drl/pdqn.rst
Original file line number Diff line number Diff line change
Expand Up @@ -8,16 +8,14 @@ PDQN_Learner
**PyTorch:**

.. py:class::
xuance.torch.learners.policy_gradient.pdqn_learner.PDQN_Learner(policy, optimizer, scheduler, summary_writer, device, model_dir, gamma, tau)
xuance.torch.learners.policy_gradient.pdqn_learner.PDQN_Learner(policy, optimizer, scheduler, device, model_dir, gamma, tau)

:param policy: The policy that provides actions and values.
:type policy: nn.Module
:param optimizer: The optimizer that update the paramters of the model.
:type optimizer: Optimizer
:param scheduler: The tool for learning rate decay.
:type scheduler: lr_scheduler
:param summary_writer: xxxxxx.
:type summary_writer: xxxxxx
:param device: The calculating device.
:type device: str
:param model_dir: The directory for saving or loading the model parameters.
Expand Down

0 comments on commit c17f588

Please sign in to comment.