Skip to content

S
seminar-breakout
Commit 677af274 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

fix build.

parent 94a445ad
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tensorpack/train/base.py
View file @ 677af274
...@@ -133,6 +133,7 @@ class Trainer(object): ...@@ -133,6 +133,7 @@ class Trainer(object):
# some final operations that might modify the graph # some final operations that might modify the graph
logger.info("Setup callbacks ...") logger.info("Setup callbacks ...")
self.config.callbacks.setup_graph(weakref.proxy(self)) self.config.callbacks.setup_graph(weakref.proxy(self))
self._extra_fetches = self.config.callbacks.extra_fetches()
if not hasattr(logger, 'LOG_DIR'): if not hasattr(logger, 'LOG_DIR'):
raise RuntimeError("logger directory wasn't set!") raise RuntimeError("logger directory wasn't set!")
......
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