- 博客(2)
- 收藏
- 关注
原创 深度学习 lstm
本人在模型里面使用 pytorch官方提供的 lstm接口的时候出现了这个状况。lstm.to('cuda) ,但是单独测试模型的时候请注释掉这段代码。只需要将lstm放进cuda里面就可以。
2023-06-02 10:26:27 117
空空如也
swintransform车牌识别问题。
2023-05-06
TA创建的收藏夹 TA关注的收藏夹
TA关注的人
本人在模型里面使用 pytorch官方提供的 lstm接口的时候出现了这个状况。lstm.to('cuda) ,但是单独测试模型的时候请注释掉这段代码。只需要将lstm放进cuda里面就可以。
2023-06-02 10:26:27 117
2023-05-06
TA创建的收藏夹 TA关注的收藏夹
TA关注的人