Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

re-trigger build (#2988) 

* fixed build

* fixed build
This commit is contained in:
William Falcon 2020-08-15 21:13:00 -04:00 committed by GitHub
parent 9f2b29a7cd
commit 766d0f391b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/base/datamodules.py
View File

@ -25,7 +25,7 @@ class TrialMNISTDataModule(LightningDataModule):
self.dims = self.mnist_train[0][0].shape
if stage == 'test' or stage is None:
self.mnist_test = TrialMNIST(root=self.data_dir, train=False, num_samples=32, download=True)
self.mnist_test = TrialMNIST(root=self.data_dir, train=False, num_samples=64, download=True)
self.dims = getattr(self, 'dims', self.mnist_test[0][0].shape)
self.non_picklable = lambda x: x**2

4
tests/models/test_amp.py
View File

@ -15,7 +15,9 @@ from tests.base import EvalModelTemplate
@pytest.mark.skipif(torch.cuda.device_count() < 2, reason="test requires multi-GPU machine")
def test_multi_gpu_wandb_ddp_spawn(tmpdir):
"""Make sure DP/DDP + AMP work."""
"""
Test ddp + wb
"""
from pytorch_lightning.loggers import WandbLogger
tutils.set_random_master_port()