提交 61b1fd00 编写于 作者: X xiexionghang

depend on paddle with bcloud

上级 72c56faf
...@@ -66,7 +66,7 @@ class EmbeddingInputLayer(Layer): ...@@ -66,7 +66,7 @@ class EmbeddingInputLayer(Layer):
emb = fluid.layers.continuous_value_model(emb, show_clk, self._cvm) emb = fluid.layers.continuous_value_model(emb, show_clk, self._cvm)
self._emb_layers.append(emb) self._emb_layers.append(emb)
output = fluid.layers.concat(input=self._emb_layers, axis=1, name=self._name) output = fluid.layers.concat(input=self._emb_layers, axis=1, name=self._name)
return output, {'data_var' : data_var} return output, {'data_var': data_var}
class LabelInputLayer(Layer): class LabelInputLayer(Layer):
...@@ -167,7 +167,8 @@ class NormalizetionLayer(Layer): ...@@ -167,7 +167,8 @@ class NormalizetionLayer(Layer):
bn = fluid.layers.data_norm(input=input_layer, name=self._name, epsilon=1e-4, param_attr={ bn = fluid.layers.data_norm(input=input_layer, name=self._name, epsilon=1e-4, param_attr={
"batch_size": 1e4, "batch_sum_default": 0.0, "batch_square": 1e4}) "batch_size": 1e4, "batch_sum_default": 0.0, "batch_square": 1e4})
inference_param = [self._name + '.batch_size', self._name + '.batch_sum', self._name + '.batch_square_sum'] inference_param = [self._name + '.batch_size', self._name + '.batch_sum', self._name + '.batch_square_sum']
return bn, {'inference_param' : {'name':'summary', 'params': inference_param, 'table_id': summary_layer.get('table_id', -1)}} return bn, {'inference_param': {'name':'summary', \
'params': inference_param, 'table_id': summary_layer.get('table_id', -1)}}
class NeuralLayer(Layer): class NeuralLayer(Layer):
...@@ -205,8 +206,9 @@ class NeuralLayer(Layer): ...@@ -205,8 +206,9 @@ class NeuralLayer(Layer):
fluid.ParamAttr(learning_rate=1.0, \ fluid.ParamAttr(learning_rate=1.0, \
initializer=fluid.initializer.NormalInitializer(loc=0.0, scale=scale)), initializer=fluid.initializer.NormalInitializer(loc=0.0, scale=scale)),
bias_attr = bias) bias_attr = bias)
inference_param = [self._name + '.w_0', self._name + '.b_0'] inference_param = [self._name + '.w_0', self._name + '.b_0']
return fc, {'inference_param' : {'name':'param', 'params': inference_param, 'table_id': param_layer.get('table_id', -1)}} return fc, {'inference_param': {'name':'param', 'params': inference_param, \
'table_id': param_layer.get('table_id', -1)}}
class SigmoidLossLayer(Layer): class SigmoidLossLayer(Layer):
...@@ -239,7 +241,6 @@ class SigmoidLossLayer(Layer): ...@@ -239,7 +241,6 @@ class SigmoidLossLayer(Layer):
} }
} }
def generate_fluid(self, param): def generate_fluid(self, param):
"""R """R
""" """
......
...@@ -10,13 +10,13 @@ import paddle.fluid as fluid ...@@ -10,13 +10,13 @@ import paddle.fluid as fluid
from paddle.fluid.incubate.fleet.parameter_server.pslib import fleet from paddle.fluid.incubate.fleet.parameter_server.pslib import fleet
def create(config): def create(config):
""" """
Create a model instance by config Create a model instance by config
Args: Args:
config(dict) : desc model type and net config(dict) : desc model type and net
Return: Return:
Model Instance Model Instance
""" """
model = None model = None
if config['mode'] == 'fluid': if config['mode'] == 'fluid':
model = FluidModel(config) model = FluidModel(config)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册