wassname/kair_algorithms_draft
1
0
Fork 0
mirror of https://github.com/wassname/kair_algorithms_draft.git synced 2026-09-09 11:25:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

Refactoring demo collector (#56)

Browse Source
This commit is contained in:
Donghyeon Kim
2019-08-17 14:29:58 +09:00
committed by Whi Kwon
parent 055d97661e
commit 18d6824191
19 changed files with 972 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/algorithms/common/abstract/agent.py
+2 -2
View File
@@ -80,7 +80,7 @@ class AbstractAgent(object):
path = os.path.join("./save/" + save_name + "_ep_" + str(n_episode) + ".pt")
torch.save(params, path)
print ("[INFO] Saved the model and optimizer to", path)
print("[INFO] Saved the model and optimizer to", path)
@abstractmethod
def write_log(self, *args):
@@ -109,7 +109,7 @@ class AbstractAgent(object):
score += reward
step += 1
print (
print(
"[INFO] episode %d\tstep: %d\ttotal score: %d"
% (i_episode, step, score)
)