diff --git a/src/main.py b/src/main.py index fadbe657f6d7ebe195ec23089078db390eecf6fd..5d3c3ea5f616ac166c88cf6c0534ec3614a14392 100644 --- a/src/main.py +++ b/src/main.py @@ -12,7 +12,7 @@ checkpoint = utility.checkpoint(args) def main(): global model - if args.data_test == 'video': + if args.data_test == ['video']: from videotester import VideoTester model = model.Model(args, checkpoint) t = VideoTester(args, model, checkpoint)