diff --git a/inference_mp4_2x.py b/inference_mp4_2x.py index 6acea8f..c8153fb 100644 --- a/inference_mp4_2x.py +++ b/inference_mp4_2x.py @@ -16,7 +16,7 @@ if torch.cuda.is_available(): parser = argparse.ArgumentParser(description='Interpolation for a pair of images') parser.add_argument('--video', dest='video', required=True) parser.add_argument('--montage', dest='montage', action='store_true', help='montage origin video') -parser.add_argument('--fps', dest='fps', type='int', default=60) +parser.add_argument('--fps', dest='fps', type=int, default=60) args = parser.parse_args() model = Model()