iwangjian / textsum-gan

Tensorflow re-implementation of GAN for text summarization
134 stars 56 forks source link

请问下生成对抗网络用于文本生成迭代次数都很高吗 #6

Open hanyegirl opened 5 years ago

hanyegirl commented 5 years ago

您好,我运行了您的代码,一天才能迭代1万次,而代码好像需要迭代100多万次,请问一下生成对抗网络用于文本这方面的需要的迭代次数或者算力都很高吗

iwangjian commented 5 years ago

GAN的训练速度本身确实很慢,后期会考虑进行代码的优化!

hanyegirl commented 5 years ago

哎,好吧。。请问下您运行这个代码需要几天呢

hanyegirl commented 5 years ago

能不能说一下您的服务器配置,我要运行的话可能需要租服务器,麻烦了

iwangjian commented 5 years ago

您可以用Titan 1080 /1080 Ti 就足够了,目前的话运行大概需要4-5天吧。

hanyegirl commented 5 years ago

您好,可不可以告诉我在哪个代码文件里设置的迭代次数啊,我找遍了所有的文件,没有找到大于100的数字。。

iwangjian commented 5 years ago

暂时还没有给出设置迭代次数的参数,后期会进行改进!

hanyegirl commented 5 years ago

暂时还没有给出设置迭代次数的参数,后期会进行改进!

那系统默认的预训练和对抗过程中的迭代次数是多少呢,系统是怎么确定什么时候停止迭代的呀,麻烦了,因为我的服务器很小,所以就想减少点迭代次数,看看代码能不能跑通

iwangjian commented 5 years ago

预训练的迭代次数代码中有体现,discriminator预训练的max_epochs=20,对抗训练的迭代过程暂没有设置迭代条件,会一直迭代进行直至手动停止。

hanyegirl commented 5 years ago

预训练的迭代次数代码中有体现,discriminator预训练的max_epochs=20,对抗训练的迭代过程暂没有设置迭代条件,会一直迭代进行直至手动停止。

真的非常感谢