Skip to content

Commit

Permalink
'a2c_agent_x(#1)'
Browse files Browse the repository at this point in the history
  • Loading branch information
baijinqiu committed Dec 24, 2023
1 parent 7dbed5f commit 5b14523
Showing 1 changed file with 11 additions and 10 deletions.
21 changes: 11 additions & 10 deletions docs/source/documents/api/learners/drl/a2c.rst
Original file line number Diff line number Diff line change
Expand Up @@ -20,11 +20,11 @@ A2C_Learner
:type device: str
:param model_dir: The directory for saving or loading the model parameters.
:type model_dir: str
:param vf_coef: xxxxxx.
:param vf_coef: Value function coefficient.
:type vf_coef: float
:param ent_coef: xxxxxx.
:param ent_coef: Entropy coefficient.
:type ent_coef: float
:param clip_grad: xxxxxx.
:param clip_grad: Gradient clipping threshold.
:type clip_grad: float

.. py:function::
Expand Down Expand Up @@ -58,11 +58,11 @@ A2C_Learner
:type device: str
:param model_dir: The directory for saving or loading the model parameters.
:type model_dir: str
:param vf_coef: xxxxxx.
:param vf_coef: Value function coefficient.
:type vf_coef: float
:param ent_coef: xxxxxx.
:param ent_coef: Entropy coefficient.
:type ent_coef: float
:param clip_grad: xxxxxx.
:param clip_grad: Gradient clipping threshold.
:type clip_grad: float

.. py:function::
Expand Down Expand Up @@ -96,13 +96,14 @@ A2C_Learner
:type scheduler: lr_scheduler
:param model_dir: The directory for saving or loading the model parameters.
:type model_dir: str
:param vf_coef: xxxxxx.
:param vf_coef: Value function coefficient.
:type vf_coef: float
:param ent_coef: xxxxxx.
:param ent_coef: Entropy coefficient.
:type ent_coef: float
:param clip_grad: xxxxxx.
:param clip_grad: Gradient clipping threshold.
:type clip_grad: float
:param clip_type: xxxxxx.
:param clip_type: Type of gradient clipping.
:type clip_type: int

.. py:function::
xuance.mindspore.learners.policy_gradient.a2c_learner.A2C_Learner.update(obs_batch, act_batch, ret_batch, adv_batch)
Expand Down

0 comments on commit 5b14523

Please sign in to comment.