From 3f5320c78f4d0793dcd5acaec8fa1a75b4a35673 Mon Sep 17 00:00:00 2001 From: Rowan McAllister Date: Tue, 20 Oct 2020 13:10:49 -0700 Subject: [PATCH] Update train.py Stack frames for CARLA pixel encoders too --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 027ee42..e69f81a 100644 --- a/train.py +++ b/train.py @@ -318,7 +318,7 @@ def main(): ) # stack several consecutive frames together - if args.encoder_type == 'pixel': + if args.encoder_type.startswith('pixel'): env = utils.FrameStack(env, k=args.frame_stack) eval_env = utils.FrameStack(eval_env, k=args.frame_stack)