Seich
/
mayavoz
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

config

This commit is contained in:
shahules786 2022-10-07 10:46:06 +05:30
parent e90efe3163
commit 61923f6d68
2 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
enhancer/cli/train_config/hyperparameters/default.yaml
View File

@ -1,7 +1,7 @@
loss : mse loss : mse
metric : mae metric : mse
lr : 0.0001 lr : 0.001
ReduceLr_patience : 5 ReduceLr_patience : 10
ReduceLr_factor : 0.1 ReduceLr_factor : 0.1
min_lr : 0.000001 min_lr : 0.000001
EarlyStopping_factor : 10 EarlyStopping_factor : 20

8
enhancer/cli/train_config/trainer/default.yaml
View File

@ -2,14 +2,14 @@ _target_: pytorch_lightning.Trainer
accelerator: auto accelerator: auto
accumulate_grad_batches: 1 accumulate_grad_batches: 1
amp_backend: native amp_backend: native
auto_lr_find: True auto_lr_find: False
auto_scale_batch_size: False auto_scale_batch_size: False
auto_select_gpus: True auto_select_gpus: True
benchmark: False benchmark: False
check_val_every_n_epoch: 1 check_val_every_n_epoch: 10
detect_anomaly: False detect_anomaly: False
deterministic: False deterministic: False
devices: auto devices: 2
enable_checkpointing: True enable_checkpointing: True
enable_model_summary: True enable_model_summary: True
enable_progress_bar: True enable_progress_bar: True
@ -23,7 +23,7 @@ limit_test_batches: 1.0
limit_train_batches: 1.0 limit_train_batches: 1.0
limit_val_batches: 1.0 limit_val_batches: 1.0
log_every_n_steps: 50 log_every_n_steps: 50
max_epochs: 10 max_epochs: 500
max_time: null max_time: null
min_epochs: 1 min_epochs: 1
min_steps: null min_steps: null