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

ensure gpu

This commit is contained in:
shahules786 2022-10-10 15:59:48 +05:30
parent 81c5f13ff6
commit a21ef707ad
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,5 +1,5 @@
_target_: pytorch_lightning.Trainer _target_: pytorch_lightning.Trainer
accelerator: auto accelerator: gpu
accumulate_grad_batches: 1 accumulate_grad_batches: 1
amp_backend: native amp_backend: native
auto_lr_find: False auto_lr_find: False
@ -9,7 +9,7 @@ benchmark: False
check_val_every_n_epoch: 1 check_val_every_n_epoch: 1
detect_anomaly: False detect_anomaly: False
deterministic: False deterministic: False
devices: 2 devices: -1
enable_checkpointing: True enable_checkpointing: True
enable_model_summary: True enable_model_summary: True
enable_progress_bar: True enable_progress_bar: True