(self, state_dim, action_dim, max_action)
| 4 | |
| 5 | class Actor(nn.Layer): |
| 6 | def __init__(self, state_dim, action_dim, max_action): |
| 7 | super(Actor, self).__init__() |
| 8 | |
| 9 | self.l1 = nn.Linear(state_dim, 400) |
| 10 | self.l2 = nn.Linear(400, 300) |
| 11 | self.l3 = nn.Linear(300, action_dim) |
| 12 | |
| 13 | self.max_action = max_action |
| 14 | |
| 15 | def forward(self, state): |
| 16 | a = F.relu(self.l1(state)) |