This commit is contained in:
johnjim0816
2021-12-22 11:19:13 +08:00
parent c257313d5b
commit 75df999258
55 changed files with 605 additions and 403 deletions

View File

@@ -33,7 +33,7 @@ class SACConfig:
self.model_path = curr_path+"/outputs/" +self.env_name+'/'+curr_time+'/models/' # path to save models
self.train_eps = 300
self.train_steps = 500
self.eval_eps = 50
self.test_eps = 50
self.eval_steps = 500
self.gamma = 0.99
self.mean_lambda=1e-3
@@ -96,7 +96,7 @@ def eval(cfg,env,agent):
print(f'Env: {cfg.env_name}, Algorithm: {cfg.algo}, Device: {cfg.device}')
rewards = []
ma_rewards = [] # moveing average reward
for i_ep in range(cfg.eval_eps):
for i_ep in range(cfg.test_eps):
state = env.reset()
ep_reward = 0
for i_step in range(cfg.eval_steps):