第二种训练中add_loss的疑问

来源:10-7 第二种模型训练(2)

慕雪1818227

2019-06-21

        dummy_encoder_inputs = np.array([
            np.array([WordSequence.PAD]) for _ in range(batch_size)])
        dummy_encoder_inputs_length = np.array([1] * batch_size)

        for epoch in range(1, n_epoch+1):
            costs = []
            bar = tqdm(range(steps), total=steps,
                       desc='epoch {}, loss=0.000000'.format(epoch))
            for _ in bar:
                x, xl, y, yl = next(flow)
                x = np.flip(x, axis=1)

                add_loss = model.train(sess,
                                       dummy_encoder_inputs,
                                       dummy_encoder_inputs_length,
                                       y, yl, loss_only=True)
                add_loss *= -0.5

老师,这一步是干嘛的,训练上述只是batch_size个的PAD?

写回答

1回答

Mr_Ricky

2019-07-01

这一部分实际上就是在做训练,训练是for那里,上面是对数据的填充。

0
0

NLP实践TensorFlow打造聊天机器人

解析自然语言处理( NLP )常用技术,从0搭建聊天机器人应用并部署上线,可用于毕设。

673 学习 · 281 问题

查看课程