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
metric : mae
lr : 0.0001
ReduceLr_patience : 5
metric : mse
lr : 0.001
ReduceLr_patience : 10
ReduceLr_factor : 0.1
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
accumulate_grad_batches: 1
amp_backend: native
auto_lr_find: True
auto_lr_find: False
auto_scale_batch_size: False
auto_select_gpus: True
benchmark: False
check_val_every_n_epoch: 1
check_val_every_n_epoch: 10
detect_anomaly: False
deterministic: False
devices: auto
devices: 2
enable_checkpointing: True
enable_model_summary: True
enable_progress_bar: True
@ -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: 10
max_epochs: 500
max_time: null
min_epochs: 1
min_steps: null