search

L1aoXingyu / Char-RNN-PyTorch

使用PyTorch实现Char RNN生成古诗和周杰伦的歌词
141 stars 41 forks source link

模型效果 #1

Closed laughing429 closed 6 years ago

laughing429 commented 6 years ago

作者您好，我这边运行了您提供的代码。训练模型的时候我使用了17MB的搜狗新闻数据，训练迭代了1000次。可是在生成文本的时候，效果并不是很好。本人也是神经网络初学者，我想请问下对于训练数据，有没有需要注意的地方呢？谢谢！

L1aoXingyu commented 6 years ago

我最近更新一下，你可以再试试

laughing429 commented 6 years ago

好的，我最近也在学习大神的源码。

laughing429 commented 6 years ago

作者您好，我这边自己训练了一个模型，可是在预测的时候，报了一个在word_to_vec中，权重文件的维度和模型的维度不一致的错。这是什么原因呢？ RuntimeError: While copying the parameter named word_to_vec.weight, whose dimensions in the model are torch.Size([2504, 512]) and whose dimensions in the checkpoint are torch.Size([2340, 512]).

L1aoXingyu commented 6 years ago

你训练的模型权重和你新定义的模型不一样，定义模型的时候需要传入文本信息来决定词向量的维度

laughing429 commented 6 years ago

哦哦，我知道了。我在训练的时候又在不断的加入训练数据，搞忘了。谢谢！