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-12-22 11:19:13 +08:00
parent c257313d5b
commit 75df999258
55 changed files with 605 additions and 403 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
codes/SAC/task0_train.ipynb
View File

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