Commit 3f1e9a14 authored by Yuxin Wu's avatar Yuxin Wu

Simple trainer accept config with no dataflow.

parent 41ba7284
...@@ -72,15 +72,14 @@ class TrainConfig(object): ...@@ -72,15 +72,14 @@ class TrainConfig(object):
if 'dataset' in kwargs: if 'dataset' in kwargs:
dataflow = kwargs.pop('dataset') dataflow = kwargs.pop('dataset')
log_deprecated("TrainConfig.dataset", "Use TrainConfig.dataflow instead.", "2017-09-11") log_deprecated("TrainConfig.dataset", "Use TrainConfig.dataflow instead.", "2017-09-11")
assert data is None or dataflow is None, "dataflow and data cannot be both presented in TrainConfig!"
if dataflow is not None: if dataflow is not None:
assert data is None, "dataflow and data cannot be both presented in TrainConfig!" assert_type(dataflow, DataFlow)
self.dataflow = dataflow
assert_type(self.dataflow, DataFlow)
self.data = None
if data is not None: if data is not None:
self.data = data assert_type(data, InputSource)
assert_type(self.data, InputSource) self.dataflow = dataflow
self.dataflow = None self.data = data
if model is not None: if model is not None:
assert_type(model, ModelDescBase) assert_type(model, ModelDescBase)
self.model = model self.model = model
......
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