diff --git a/distiller/utils.py b/distiller/utils.py index f803d171b70378815998515c1687bd5eaa62dbb6..13d1f4aad0c2431339a0d2caa7370479ca48a00e 100755 --- a/distiller/utils.py +++ b/distiller/utils.py @@ -30,12 +30,6 @@ def to_np(var): return var.data.cpu().numpy() -def to_var(tensor, cuda=True): - if cuda and torch.cuda.is_available(): - tensor = tensor.cuda() - return Variable(tensor) - - def size2str(torch_size): if isinstance(torch_size, torch.Size): return size_to_str(torch_size)