Minor Changes

2023-05-22 14:25:26 +02:00 · 2023-05-22 14:25:26 +02:00 · f28db0ffe3
commit f28db0ffe3
parent c80588fcbb
1 changed files with 4 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -146,7 +146,10 @@ def main():
        from_pixels=(args.encoder_type == 'pixel'),
        height=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)