inputs are weird

Raise code

def to_device(i, d):
    if isinstance(i, torch.Tensor):
        return i.to(device=d)
    elif isinstance(i, (tuple, list)):
        return tuple(to_device(e, d) for e in i)
    else:
        raise RuntimeError('inputs are weird')

class GPUWrapper(torch.nn.Module):
    def __init__(self, root):
        super().__init__()
        self.models = []
        self.streams = {}
        for i in range(torch.cuda.device_count()):