提交 fc0e6988 编写于 作者: D dengkaipeng

remove useless code in AttentionCluster

上级 a9203822
[MODEL]
name = "AttentionCluster"
dataset = "YouTube-8M"
bone_nework = None
bone_network = None
drop_rate = 0.5
feature_num = 2
feature_names = ['rgb', 'audio']
......
......@@ -3,12 +3,10 @@ import paddle.fluid as fluid
class LogisticModel(object):
"""Logistic model with L2 regularization."""
"""Logistic model."""
def build_model(self,
model_input,
vocab_size,
l2_penalty=None,
**unused_params):
"""Creates a logistic model.
......
......@@ -18,7 +18,6 @@ class ShiftingAttentionModel(object):
x_shape.stop_gradient = True
flat_x = fluid.layers.reshape(x, shape=(-1, self.seg_num))
flat_softmax = fluid.layers.softmax(flat_x)
# return fluid.layers.reshape(flat_softmax, shape=(x.shape[0], self.n_att, -1), actual_shape=x_shape)
return fluid.layers.reshape(
flat_softmax, shape=x.shape, actual_shape=x_shape)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册