From 55910e61070ce9679bd85cd63695f04c5ed4fb9f Mon Sep 17 00:00:00 2001 From: Thalles Silva Date: Thu, 21 Jan 2021 06:32:50 -0300 Subject: [PATCH] fix loss function labels --- run.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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')