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

fix TrainConfig on default arguments

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