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')