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

Merge pull request #3 from rowanmcallister/patch-1

Browse Source 
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
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)