From 6426c7b818dfbef97f85e686444b8ef77cf352d6 Mon Sep 17 00:00:00 2001 From: Eren Golge 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):