From f28db0ffe372f53a86a7e55212a7604622bedd47 Mon Sep 17 00:00:00 2001 From: VedantDave Date: Mon, 22 May 2023 14:25:26 +0200 Subject: [PATCH] Minor Changes --- train.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/train.py b/train.py index 02769f7..047f1ee 100644 --- 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)