hongshaorou/Tianshou
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix tqdm issue (#481)

Browse Source
This commit is contained in:
Jiayi Weng 2021-11-18 11:17:44 -05:00 committed by GitHub
parent 8f19a86966
commit 94d3b27db9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tianshou/trainer/offpolicy.py
View File

@ -117,8 +117,8 @@ def offpolicy_trainer(
env_step += int(result["n/st"]) env_step += int(result["n/st"])
t.update(result["n/st"]) t.update(result["n/st"])
logger.log_train_data(result, env_step) logger.log_train_data(result, env_step)
last_rew = result['rew'] if 'rew' in result else last_rew last_rew = result['rew'] if result["n/ep"] > 0 else last_rew
last_len = result['len'] if 'len' in result else last_len last_len = result['len'] if result["n/ep"] > 0 else last_len
data = { data = {
"env_step": str(env_step), "env_step": str(env_step),
"rew": f"{last_rew:.2f}", "rew": f"{last_rew:.2f}",

4
tianshou/trainer/onpolicy.py
View File

@ -125,8 +125,8 @@ def onpolicy_trainer(
env_step += int(result["n/st"]) env_step += int(result["n/st"])
t.update(result["n/st"]) t.update(result["n/st"])
logger.log_train_data(result, env_step) logger.log_train_data(result, env_step)
last_rew = result['rew'] if 'rew' in result else last_rew last_rew = result['rew'] if result["n/ep"] > 0 else last_rew
last_len = result['len'] if 'len' in result else last_len last_len = result['len'] if result["n/ep"] > 0 else last_len
data = { data = {
"env_step": str(env_step), "env_step": str(env_step),
"rew": f"{last_rew:.2f}", "rew": f"{last_rew:.2f}",