码迷,mamicode.com
首页 > 其他好文 > 详细

tf 版本更新 记录

时间:2019-03-18 10:25:16      阅读:208      评论:0      收藏:0      [点我收藏+]

标签:代码   str   out   class   版本   seq   drop   tmc   tar   

tf 经常更新版本,网上教程又是各版本都有,且不标明版本,致使各种用法难以分清哪个新,哪个旧,这里做个记录,以前的博客我就不更新了,请大家见谅。

 

tf.nn.rnn_cell 改为 tf.contrib.rnn

# 原代码
lstm_cell = tf.nn.rnn_cell.BasicLSTMCell(HIDDEN_SIZE)
if is_training:
    lstm_cell = tf.nn.rnn_cell.DropoutWrapper(lstm_cell, output_keep_prob=KEEP_PROB)
cell = tf.nn.rnn_cell.MultiRNNCell([lstm_cell] * NUM_LAYERS)

# 修改为
lstm_cell = tf.contrib.rnn.BasicLSTMCell(HIDDEN_SIZE)
if is_training:
    lstm_cell = tf.contrib.rnn.DropoutWrapper(lstm_cell, output_keep_prob=KEEP_PROB)
cell = tf.contrib.rnn.MultiRNNCell([lstm_cell] * NUM_LAYERS)

 

tf.nn.seq2seq.sequence_loss_by_example 改为 tf.contrib.legacy_seq2seq.sequence_loss_by_example

# 原代码
loss = tf.nn.seq2seq.sequence_loss_by_example(
            [logits],  [tf.reshape(self.targets, [-1])], 
            [tf.ones([batch_size * num_steps], dtype=tf.float32)])

# 修改为
loss = tf.contrib.legacy_seq2seq.sequence_loss_by_example(
            [logits],  [tf.reshape(self.targets, [-1])], 
            [tf.ones([batch_size * num_steps], dtype=tf.float32)])

 

tf.concat

# 原代码
concated = tf.concat(1, [indices, labels])
# 修改为
concated = tf.concat([indices, labels], 1)

 

tf 版本更新 记录

标签:代码   str   out   class   版本   seq   drop   tmc   tar   

原文地址:https://www.cnblogs.com/yanshw/p/10550597.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!