log saver added

CUN-bjy · Nov 22, 2020 · b9003a1 · b9003a1
1 parent da41852
commit b9003a1
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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')