def build_model_graph(config, params, dataset):
graph = tf.Graph()
with graph.as_default():
iterators = dataset.make_tf_iterators(params)
model = BOWFlipper(config, params, dataset, iterators)
return tf_utils.TFModel(graph=graph, model=model, iterator=iterators)
def build_model_graph(config, params, dataset):
graph = tf.Graph()
with graph.as_default():
iterators = dataset.make_tf_iterators(params['batch_size'])
model = ABow(config, params, dataset, iterators)
return tf_utils.TFModel(graph=graph, model=model, iterator=iterators)