igorcheb commited on
Commit
a3988fe
1 Parent(s): 38a4bc3

Update Agent_class.py

Browse files
Files changed (1) hide show
  1. Agent_class.py +2 -3
Agent_class.py CHANGED
@@ -4,9 +4,8 @@ class ParameterisedPolicy(torch.nn.Module):
4
  """
5
  REINFORCE RL agent class. Returns action when the ParameterisedPolicy.act(observation) method is used.
6
  observation is a gym state vector.
7
- obs_len - length of the state vector
8
- act_space_len - length of the action vector
9
-
10
  """
11
  def __init__(self, obs_len=8, act_space_len=2):
12
  super().__init__()
 
4
  """
5
  REINFORCE RL agent class. Returns action when the ParameterisedPolicy.act(observation) method is used.
6
  observation is a gym state vector.
7
+ obs_len - length of the state vector.
8
+ act_space_len - length of the action vector.
 
9
  """
10
  def __init__(self, obs_len=8, act_space_len=2):
11
  super().__init__()