Update perspective.py

fix logdir bug inFeatureVisualization
dingguanglei · Dec 5, 2019 · dacab5c · dacab5c
1 parent 416ec21
commit dacab5c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/jdit/perspective.py b/jdit/perspective.py
@@ -19,7 +19,7 @@ def __init__(self, model: Union[Model, Module], logdir: str):
         model.eval()
         self.model = model
         self.activations: Dict[str, Tensor] = {}
-        self.watcher = Watcher(logdir=log)
+        self.watcher = Watcher(logdir=logdir)
 
     def _hook(self, name: str):
         def hook(model, input, output):