From 845575a2ad4cac236c5e15c0e967acb3f1b38962 Mon Sep 17 00:00:00 2001 From: shahules786 Date: Sat, 8 Oct 2022 10:18:22 +0530 Subject: [PATCH] config --- enhancer/cli/train_config/trainer/default.yaml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/enhancer/cli/train_config/trainer/default.yaml b/enhancer/cli/train_config/trainer/default.yaml index c7b0417..602cbc2 100644 --- a/enhancer/cli/train_config/trainer/default.yaml +++ b/enhancer/cli/train_config/trainer/default.yaml @@ -6,7 +6,7 @@ auto_lr_find: False auto_scale_batch_size: False auto_select_gpus: True benchmark: False -check_val_every_n_epoch: 10 +check_val_every_n_epoch: 1 detect_anomaly: False deterministic: False devices: 2 @@ -23,7 +23,7 @@ limit_test_batches: 1.0 limit_train_batches: 1.0 limit_val_batches: 1.0 log_every_n_steps: 50 -max_epochs: 500 +max_epochs: 10 max_time: null min_epochs: 1 min_steps: null