From ed30840f4e6a842143b279cedb8bbd0e57e34057 Mon Sep 17 00:00:00 2001 From: shahules786 Date: Wed, 19 Oct 2022 16:41:23 +0530 Subject: [PATCH] cpu --- enhancer/cli/train_config/trainer/default.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/enhancer/cli/train_config/trainer/default.yaml b/enhancer/cli/train_config/trainer/default.yaml index 4fa4438..bd51d8a 100644 --- a/enhancer/cli/train_config/trainer/default.yaml +++ b/enhancer/cli/train_config/trainer/default.yaml @@ -1,5 +1,5 @@ _target_: pytorch_lightning.Trainer -accelerator: gpu +accelerator: cpu accumulate_grad_batches: 1 amp_backend: native auto_lr_find: False