Attention-Seq2Seq-Chatbot-by-Pytorch1.0.1
Attention-Seq2Seq-Chatbot-by-Pytorch1.0.1 copied to clipboard
训练何时停止?
训练了一晚上了,还在跑   代码就这么几行 ``` dataClass = Corpus('./corpus/chat.tsv', maxSentenceWordsNum=25,testSize=1) model = Seq2Seq(dataClass, featureSize=256, hiddenSize=256, attnType='L', attnMethod='general', encoderNumLayers=3, decoderNumLayers=2, encoderBidirectional=True, device=torch.device('cpu')) model.train(batchSize=1024, epoch=500) model.save('model.pkl') ```
``` ValueError: test_size=0.0 should be either positive and smaller than the number of samples 256578 or a float in the (0, 1) range ``` 这是什么原因??
可视化的问题
请教下这个项目用tensorboard如何实现可视化呢?