From b29c8020d7ac72256dca48fae85e96b4b3c75ccb Mon Sep 17 00:00:00 2001 From: pzhokhov Date: Thu, 30 Aug 2018 13:40:40 -0700 Subject: [PATCH] remove saving model as a pickle file in ppo2 (tries to pull environment in; bad idea - may need to use constructor argument pickling or somesuch if at all necessary) (#69) --- baselines/ppo2/ppo2.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/baselines/ppo2/ppo2.py b/baselines/ppo2/ppo2.py index 88b5b3c..72c9289 100644 --- a/baselines/ppo2/ppo2.py +++ b/baselines/ppo2/ppo2.py @@ -227,10 +227,6 @@ def learn(*, network, env, total_timesteps, seed=None, nsteps=2048, ent_coef=0.0 make_model = lambda : Model(policy=policy, ob_space=ob_space, ac_space=ac_space, nbatch_act=nenvs, nbatch_train=nbatch_train, nsteps=nsteps, ent_coef=ent_coef, vf_coef=vf_coef, max_grad_norm=max_grad_norm) - if save_interval and logger.get_dir(): - import cloudpickle - with open(osp.join(logger.get_dir(), 'make_model.pkl'), 'wb') as fh: - fh.write(cloudpickle.dumps(make_model)) model = make_model() if load_path is not None: model.load(load_path)