bug fix

2026-09-09 11:16:00 +08:00 · 2018-04-25 05:38:23 -07:00
parent 03033a99ef
commit 6426c7b818
1 changed files with 1 additions and 0 deletions
@@ -9,6 +9,7 @@ import torch
 import subprocess
 import numpy as np
 from collections import OrderedDict
+from torch.autograd import Variable


 class AttrDict(dict):