From 6968572c1a21394b88a29f675b17b9698784a898 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Fri, 25 Aug 2023 23:19:39 +0200
Subject: Update training stuff

---
 training/conf/trainer/default.yaml | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

(limited to 'training/conf/trainer/default.yaml')

diff --git a/training/conf/trainer/default.yaml b/training/conf/trainer/default.yaml
index 6112cd8..2e593e8 100644
--- a/training/conf/trainer/default.yaml
+++ b/training/conf/trainer/default.yaml
@@ -1,14 +1,12 @@
 _target_: pytorch_lightning.Trainer
-auto_scale_batch_size: binsearch
-auto_lr_find: false
 gradient_clip_val: 0.5
 fast_dev_run: false
-gpus: 1
+accelerator: gpu
+devices: 1
 precision: 16
 max_epochs: 256
 limit_train_batches: 1.0
 limit_val_batches: 1.0
 limit_test_batches: 1.0
-resume_from_checkpoint: null
 accumulate_grad_batches: 1
 overfit_batches: 0
-- 
cgit v1.2.3-70-g09d2