Commit 9071202f authored by ly015's avatar ly015 Committed by Wenwei Zhang
Browse files

fix PaviLoggerHook

parent 792473c9
...@@ -44,7 +44,7 @@ class PaviLoggerHook(LoggerHook): ...@@ -44,7 +44,7 @@ class PaviLoggerHook(LoggerHook):
if not self.init_kwargs: if not self.init_kwargs:
self.init_kwargs = dict() self.init_kwargs = dict()
self.init_kwargs['task'] = self.run_name self.init_kwargs['name'] = self.run_name
self.init_kwargs['model'] = runner._model_name self.init_kwargs['model'] = runner._model_name
if runner.meta is not None: if runner.meta is not None:
if 'config_dict' in runner.meta: if 'config_dict' in runner.meta:
...@@ -112,3 +112,7 @@ class PaviLoggerHook(LoggerHook): ...@@ -112,3 +112,7 @@ class PaviLoggerHook(LoggerHook):
image = data[self.img_key][0:1].to(device) image = data[self.img_key][0:1].to(device)
with torch.no_grad(): with torch.no_grad():
self.writer.add_graph(_model, image) self.writer.add_graph(_model, image)
def __del__(self):
if hasattr(self, 'writer'):
self.writer.close()
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment