mirror of
https://github.com/wassname/kair_algorithms_draft.git
synced 2026-09-09 11:25:10 +08:00
Modified demo config and run demo script (#83)
* Add removed q_dot scaling factor * Refactored cubic function * Separate config and add run file. * Resolve flake8 issue * Resolve flake8 issue * Remove whitespace before ( issue
This commit is contained in:
@@ -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)
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user