diff --git a/run.py b/run.py index de3b0eb..03606a5 100644 --- a/run.py +++ b/run.py @@ -54,7 +54,7 @@ parser.add_argument('--gpu-index', default=0, type=int, help='Gpu index.') def main(): args = parser.parse_args() - assert args.n_views == 2, "Only two view training is supported." + assert args.n_views == 2, "Only two view training is supported. Please use --n-views 2." # check if gpu training is available if not args.disable_cuda and torch.cuda.is_available(): args.device = torch.device('cuda')