diff --git a/pytorch_lightning/pt_overrides/override_data_parallel.py b/pytorch_lightning/pt_overrides/override_data_parallel.py index 89b550fdbb..f06f8030ad 100644 --- a/pytorch_lightning/pt_overrides/override_data_parallel.py +++ b/pytorch_lightning/pt_overrides/override_data_parallel.py @@ -6,7 +6,6 @@ from itertools import chain import threading import torch from torch.cuda._utils import _get_device_index -import pdb def _find_tensors(obj): # pragma: no cover diff --git a/pytorch_lightning/root_module/grads.py b/pytorch_lightning/root_module/grads.py index 8ed17a3eec..e368bb5f50 100644 --- a/pytorch_lightning/root_module/grads.py +++ b/pytorch_lightning/root_module/grads.py @@ -1,4 +1,3 @@ -import numpy as np from torch import nn """