removing unused imports
This commit is contained in:
parent
40e5f1b4c8
commit
f2ef367f7d
|
@ -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
|
||||
|
|
|
@ -1,4 +1,3 @@
|
|||
import numpy as np
|
||||
from torch import nn
|
||||
|
||||
"""
|
||||
|
|
Loading…
Reference in New Issue