Commit 11a9650f authored by Yuxin Wu's avatar Yuxin Wu

fix TrainConfig on default arguments

parent 0a13579c
...@@ -105,10 +105,12 @@ class TrainConfig(object): ...@@ -105,10 +105,12 @@ class TrainConfig(object):
if callbacks is None: if callbacks is None:
callbacks = [] callbacks = []
assert_type(callbacks, list) assert_type(callbacks, list)
self._callbacks = callbacks + \ if extra_callbacks is not None:
(extra_callbacks or DEFAULT_CALLBACKS()) self._callbacks = callbacks + extra_callbacks
else:
self._callbacks = callbacks + DEFAULT_CALLBACKS()
self.monitors = monitors or DEFAULT_MONITORS() self.monitors = monitors if monitors is not None else DEFAULT_MONITORS()
if session_init is None: if session_init is None:
session_init = JustCurrentSession() session_init = JustCurrentSession()
......
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