From b9003a1d4290a0f2b37a20b6379fb9aeed3fa9bb Mon Sep 17 00:00:00 2001 From: CUN-bjy Date: Sun, 22 Nov 2020 12:48:05 +0900 Subject: [PATCH] log saver added --- train.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/train.py b/train.py index a0dacec..372fc6f 100644 --- a/train.py +++ b/train.py @@ -135,6 +135,10 @@ def model_train(pretrained_): agent.save_weights(path +'temp') env.close() + # log saver + import pickle + pickle.dump(open(path+'%s.log'%time.Time.now(),'wb')) + argparser = argparse.ArgumentParser( description='Train DDPG Agent on the openai gym')