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