pt_chatbot

pt_chatbot copied to clipboard

https://github.com/coranholmes/pt_chatbot/blob/b6e7416e4787542ed07f8b40baafde57738fffaa/model.py#L213 使用pytorch 1.4.0会跳出 Warning: masked_scatter_ received a mask with dtype torch.uint8, this behavior is now deprecated,please use a mask with dtype torch.bool instead. (masked_scatter__cuda at ..\aten\src\ATen\native\cuda\LegacyDefinitions.cpp:57) 修改成这样就可解决 mask = torch.ByteTensor(mask).bool()

InPRTx

为啥运行了train.py 出现： [Running] python -u "e:\VScode\Python\chatbot\pt_chatbot-master\pt_chatbot-master\train.py" Building encoder and decoder ... Models built and ready to go! Evaluation Mode starts ... Building prefix dict from the default dictionary ... Loading...

dahudieyu