From d6a6935366ee19550671236f77361d6d4ab4f303 Mon Sep 17 00:00:00 2001
From: Eren Golge <egolge@mozilla.com>
Date: Wed, 25 Apr 2018 05:38:23 -0700
Subject: [PATCH] bug fix

---
 utils/generic_utils.py | 1 +
 1 file changed, 1 insertion(+)

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):