Merge pull request #3 from rowanmcallister/patch-1

Update train.py
This commit is contained in:
amyzhang 2020-10-20 13:12:01 -07:00 committed by GitHub
commit 15008bbd3b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -318,7 +318,7 @@ def main():
) )
# stack several consecutive frames together # stack several consecutive frames together
if args.encoder_type == 'pixel': if args.encoder_type.startswith('pixel'):
env = utils.FrameStack(env, k=args.frame_stack) env = utils.FrameStack(env, k=args.frame_stack)
eval_env = utils.FrameStack(eval_env, k=args.frame_stack) eval_env = utils.FrameStack(eval_env, k=args.frame_stack)