From 40b86808c86fd13f7bedae47a61e4b626ca2bf48 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 24 Jul 2019 18:32:48 -0400 Subject: [PATCH] running ddp tests --- pytorch_lightning/pt_overrides/override_data_parallel.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/pytorch_lightning/pt_overrides/override_data_parallel.py b/pytorch_lightning/pt_overrides/override_data_parallel.py index 9b287de1..9d34d455 100644 --- a/pytorch_lightning/pt_overrides/override_data_parallel.py +++ b/pytorch_lightning/pt_overrides/override_data_parallel.py @@ -9,7 +9,7 @@ from torch.cuda._utils import _get_device_index import pdb -def _find_tensors(obj): +def _find_tensors(obj): # pragma: no cover r""" Recursively find all tensors contained in the specified object. """ @@ -22,8 +22,7 @@ def _find_tensors(obj): return [] - -def get_a_var(obj): +def get_a_var(obj): # pragma: no cover if isinstance(obj, torch.Tensor): return obj