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)