mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-10 12:21:57 +08:00
Fix failing GPU tests (#722)
* Fix distributed_backend=None test We now throw a warning instead of an exception. Update test to reflect this. * Fix test_tube logger close when debug=True
This commit is contained in:
committed by
William Falcon
parent
ca894f081b
commit
dfb6d3626e
@@ -135,8 +135,9 @@ class TestTubeLogger(LightningLoggerBase):
|
||||
def close(self):
|
||||
# TODO: HACK figure out where this is being set to true
|
||||
self.experiment.debug = self.debug
|
||||
exp = self.experiment
|
||||
exp.close()
|
||||
if not self.debug:
|
||||
exp = self.experiment
|
||||
exp.close()
|
||||
|
||||
@property
|
||||
def rank(self):
|
||||
|
||||
@@ -183,7 +183,7 @@ def test_multi_gpu_none_backend(tmpdir):
|
||||
gpus='-1'
|
||||
)
|
||||
|
||||
with pytest.raises(MisconfigurationException):
|
||||
with pytest.warns(UserWarning):
|
||||
tutils.run_model_test(trainer_options, model)
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user