From ea0db1f9957c86f925ab3af4b182d1e988548430 Mon Sep 17 00:00:00 2001 From: Mike Walmsley Date: Tue, 7 Nov 2023 11:50:07 -0500 Subject: [PATCH] 2 gpu --- only_for_me/narval/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/only_for_me/narval/train.py b/only_for_me/narval/train.py index 3af6adaa..e5c8bbcb 100644 --- a/only_for_me/narval/train.py +++ b/only_for_me/narval/train.py @@ -118,7 +118,7 @@ resize_after_crop=args.resize_after_crop, # hardware parameters # gpus=args.gpus, - gpus=4, + gpus=2, nodes=args.nodes, mixed_precision=args.mixed_precision, wandb_logger=wandb_logger,