search

FrankWork / fudan_mtl_reviews

TensorFlow implementation of the paper `Adversarial Multi-task Learning for Text Classification`
178 stars 40 forks source link

关于bach数目设定的问题 #19

Open Haiming94 opened 4 years ago

Haiming94 commented 4 years ago

您好，我已经看这份代码很多遍了，今天突然发现一个问题，我主要想知道下面这个循环：

for batch in range(82): train_fetch = [m_train.tensors, m_train.train_ops, merged_train]

其中，“82”是如何得到的呢？ 谢谢，希望您能和我交流！

weidalan commented 4 years ago

路过回答一下哈。数据是从dataset里流入的，复制了epoch次。默认的batch是16，所以这个地方只要满足 batch * range的范围 < 最少的训练集的句子数 就可以。