From 65d6b1815988cbecb4d70a6c61b8e4063a0efab6 Mon Sep 17 00:00:00 2001 From: Pablo Millan Date: Mon, 25 Mar 2024 11:21:21 -0400 Subject: [PATCH] Update __main__.py BUG: Compatibility with Google Colab --- idelucs/__main__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/idelucs/__main__.py b/idelucs/__main__.py index 1e599ef..516effd 100644 --- a/idelucs/__main__.py +++ b/idelucs/__main__.py @@ -312,7 +312,7 @@ def main(): for key in args: print(f'{key} \t -> {args[key]}') - torch.set_num_threads(cpu_count() - 2 ) + torch.set_num_threads(cpu_count()) run(args)