diff --git a/utils/generic_utils.py b/utils/generic_utils.py
index 33508f9e..0103b169 100644
--- a/utils/generic_utils.py
+++ b/utils/generic_utils.py
@@ -9,6 +9,7 @@ import torch
 import subprocess
 import numpy as np
 from collections import OrderedDict
+from torch.autograd import Variable
 
 
 class AttrDict(dict):