wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

testing -1 gpu option

Browse Source
This commit is contained in:
William Falcon
2019-07-24 19:02:19 -04:00
parent f58c83b399
commit 65ce10c255
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_models.py
+1 -1
View File
@@ -195,7 +195,7 @@ def test_multi_gpu_model_dp():
max_nb_epochs=1,
train_percent_check=0.1,
val_percent_check=0.1,
gpus=[0, 1]
gpus='-1'
)
run_gpu_model_test(trainer_options, model, hparams)