From a1cef656b848f6fb6c6bd4920d71191c616536d8 Mon Sep 17 00:00:00 2001 From: peter Date: Thu, 1 Nov 2018 12:10:50 -0700 Subject: [PATCH] pass microbatch_size to the model during construction --- baselines/ppo2/ppo2.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/baselines/ppo2/ppo2.py b/baselines/ppo2/ppo2.py index cf4a24c..a0adadc 100644 --- a/baselines/ppo2/ppo2.py +++ b/baselines/ppo2/ppo2.py @@ -43,7 +43,10 @@ class Model(object): act_model = policy(nbatch_act, 1, sess) # Train model for training - train_model = policy(nbatch_train, nsteps, sess) + if microbatch_size is None: + train_model = policy(nbatch_train, nsteps, sess) + else: + train_model = policy(microbatch_size, nsteps, sess) # CREATE THE PLACEHOLDERS A = train_model.pdtype.sample_placeholder([None])