pytorch 踩坑之'tuple' object has no attribute 'size'

博主因使用BERT,需将Keras代码移植到PyTorch。写BiLSTM时遇到tuple报错问题,向大佬请教后发现是pytorch summary的bug。因pytorch写LSTM时output是tuple,含hn、cn,而summary无法识别tuple导致报错。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

最近因为要用BERT的原因,需要将之前keras写的一些代码,移植到pytorch上,在写BiLSTM的时候,遇到一个问题,

一脸懵逼,哪里来的tuple!!!疯了,后来,弱小无助的小菜鸡终于忍不住向大佬请教,然后大佬也疯了,最终发现这是pytorch summary的一个bug,因为pytorch在写LSTM的时候的output是一个tuple,除了output之外还有所谓的hn,cn然而summary无法识别tuple,所以才会报这个错误。

评论 43
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值