import tensorflow as tf

        if FLAGS.task.reset_policy:
            # NOTE: reset policy and valuefunc
            logger.info("Resetting Policy")
            pol_params = tf.get_default_session().run([nn.utils.parameters_to_vector(policy.parameters())])
            tf.get_default_session().run(tf.variables_initializer(policy.parameters()))
            pol_params_after = tf.get_default_session().run([nn.utils.parameters_to_vector(policy.parameters())])
            print ("pol_params:", np.linalg.norm(pol_params), "pol_params_after_reset:", np.linalg.norm(pol_params_after))
            logger.info("Resetting Valuefunc")
            tf.get_default_session().run(tf.variables_initializer(vfn.parameters()))

            tf.get_default_session().run(tf.variables_initializer(warmup_policy.parameters()))
            tf.get_default_session().run(tf.variables_initializer(warmup_vfn.parameters()))
            for p in warmup_policy.parameters(): p.invalidate()
            for p in warmup_vfn.parameters(): p.invalidate()
            for p in policy.parameters(): p.invalidate()
            for p in vfn.parameters(): p.invalidate()
