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

Minor Changes

Browse Source
This commit is contained in:
Vedant Dave 2023-05-22 14:25:26 +02:00
parent c80588fcbb
commit f28db0ffe3
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
train.py
View File

@ -146,7 +146,10 @@ def main():
from_pixels=(args.encoder_type == 'pixel'), from_pixels=(args.encoder_type == 'pixel'),
height=args.image_size, height=args.image_size,
width=args.image_size, width=args.image_size,
frame_skip=args.action_repeat frame_skip=args.action_repeat,
img_source=args.img_source,
resource_files=args.resource_files,
total_frames=args.total_frames
) )
env.seed(args.seed) env.seed(args.seed)