ensure 2 gpus
This commit is contained in:
parent
f66a5236e1
commit
5a392332ba
|
|
@ -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: -1
|
devices: 2
|
||||||
enable_checkpointing: True
|
enable_checkpointing: True
|
||||||
enable_model_summary: True
|
enable_model_summary: True
|
||||||
enable_progress_bar: True
|
enable_progress_bar: True
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue