added test for no dist sampler

This commit is contained in:
William Falcon
2019-07-24 17:11:25 -04:00
parent d1d33e8db6
commit b30fbf80d0
+6
View File
@@ -37,7 +37,11 @@ def test_ddp_sampler_error():
hparams = get_hparams()
model = LightningTestModel(hparams, force_remove_distributed_sampler=True)
exp = get_exp(True)
exp.save()
trainer = Trainer(
experiment=exp,
progress_bar=False,
max_nb_epochs=1,
gpus=[0, 1],
@@ -48,6 +52,8 @@ def test_ddp_sampler_error():
with pytest.raises(MisconfigurationException):
trainer.get_dataloaders(model)
clear_save_dir()
#
# def test_cpu_model():
# """