From 251173ffa9f87da96df91af71ec600d5e1503efd Mon Sep 17 00:00:00 2001 From: Somdev Sangwan Date: Tue, 30 May 2023 07:01:02 +0530 Subject: [PATCH] bruh moment 9 --- run.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/run.py b/run.py index aceab00..d94ff6a 100755 --- a/run.py +++ b/run.py @@ -42,7 +42,7 @@ parser.add_argument('--keep-frames', help='keep frames directory', dest='keep_fr for name, value in vars(parser.parse_args()).items(): args[name] = value -if torch.cuda.is_available() not args['gpu']: +if not torch.cuda.is_available() and args['gpu']: quit("You are using --gpu flag but CUDA isn't available on your system.") sep = "/"