VedantDave/DBC
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update train.py

Browse Source 
Stack frames for CARLA pixel encoders too
This commit is contained in:
Rowan McAllister 2020-10-20 13:10:49 -07:00 committed by GitHub
parent 0fc1b8bd37
commit 3f5320c78f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
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)