bacow/easy-rl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
JohnJim0816
2021-03-23 16:10:11 +08:00
parent d4690c2058
commit bf0f2990cf
198 changed files with 1668 additions and 1545 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
codes/A2C/agent.py
View File

@@ -5,19 +5,18 @@ Author: John
Email: johnjim0816@gmail.com
Date: 2020-11-03 20:47:09
LastEditor: John
LastEditTime: 2020-11-08 22:16:29
LastEditTime: 2021-03-20 17:41:21
Discription:
Environment:
'''
from model import ActorCritic
from A2C.model import ActorCritic
import torch.optim as optim
class A2C:
def __init__(self,n_states, n_actions, hidden_dim=256,device="cpu",lr = 3e-4):
self.device = device
def __init__(self,n_states, n_actions, cfg):
self.gamma = 0.99
self.model = ActorCritic(n_states, n_actions, hidden_dim=hidden_dim).to(device)
self.optimizer = optim.Adam(self.model.parameters(),lr=lr)
self.model = ActorCritic(n_states, n_actions, hidden_dim=cfg.hidden_dim).to(cfg.device)
self.optimizer = optim.Adam(self.model.parameters(),lr=cfg.lr)
def choose_action(self, state):
dist, value = self.model(state)
action = dist.sample()