Remove tf logger

2026-09-11 11:53:01 +08:00 · 2018-05-19 11:31:56 -06:00
parent d846bc50d7
commit 28a9e41a08
2 changed files with 1 additions and 62 deletions
@@ -41,9 +41,7 @@ class Logger(object):
        self.all_steps = {}

    def to_numpy(self, v):
-        if isinstance(v, torch.autograd.Variable):
-            v = v.data
-        if isinstance(v, torch.FloatTensor):
+        if isinstance(v, torch.Tensor):
            v = v.cpu().detach().numpy()
        return v