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

added test for no dist sampler

Browse Source
This commit is contained in:
William Falcon
2019-07-24 17:02:39 -04:00
parent 1e0bae14da
commit 9e5dd7a7ea
1 changed files with 0 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_models.py
-9
View File
@@ -231,19 +231,10 @@ def test_amp_gpu_ddp():
run_gpu_model_test(trainer_options, model, hparams)
# ------------------------------------------------------------------------
# UTILS
# ------------------------------------------------------------------------
def run_gpu_model_test(trainer_options, model, hparams, on_gpu=True):
"""
Make sure DDP + AMP work
:return:
"""
save_dir = init_save_dir()
# exp file to get meta