最近因为要用BERT的原因,需要将之前keras写的一些代码,移植到pytorch上,在写BiLSTM的时候,遇到一个问题,
一脸懵逼,哪里来的tuple!!!疯了,后来,弱小无助的小菜鸡终于忍不住向大佬请教,然后大佬也疯了,最终发现这是pytorch summary的一个bug,因为pytorch在写LSTM的时候的output是一个tuple,除了output之外还有所谓的hn,cn然而summary无法识别tuple,所以才会报这个错误。
最近因为要用BERT的原因,需要将之前keras写的一些代码,移植到pytorch上,在写BiLSTM的时候,遇到一个问题,
一脸懵逼,哪里来的tuple!!!疯了,后来,弱小无助的小菜鸡终于忍不住向大佬请教,然后大佬也疯了,最终发现这是pytorch summary的一个bug,因为pytorch在写LSTM的时候的output是一个tuple,除了output之外还有所谓的hn,cn然而summary无法识别tuple,所以才会报这个错误。