From aa043aaf405da56178d5e19cdfed4542b5f17d62 Mon Sep 17 00:00:00 2001 From: shahules786 Date: Thu, 6 Oct 2022 11:52:05 +0530 Subject: [PATCH] rmv max_steps --- enhancer/cli/train_config/trainer/default.yaml | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/enhancer/cli/train_config/trainer/default.yaml b/enhancer/cli/train_config/trainer/default.yaml index 515938f..7c387bc 100644 --- a/enhancer/cli/train_config/trainer/default.yaml +++ b/enhancer/cli/train_config/trainer/default.yaml @@ -9,7 +9,7 @@ benchmark: False check_val_every_n_epoch: 1 detect_anomaly: False deterministic: False -devices: -1 +devices: auto enable_checkpointing: True enable_model_summary: True enable_progress_bar: True @@ -24,7 +24,6 @@ limit_train_batches: 1.0 limit_val_batches: 1.0 log_every_n_steps: 50 max_epochs: 10 -max_steps: null max_time: null min_epochs: 1 min_steps: null