pass microbatch_size to the model during construction

2018-11-01 12:10:50 -07:00
parent b0589da817
commit a1cef656b8
1 changed files with 4 additions and 1 deletions
--- 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])