expected type torch.cuda.DoubleTensor but got torch.cuda.FloatTensor python报错

川长思鸟来 2022-02-04 05:11 349阅读 0赞

watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM5OTM4NjY2_size_16_color_FFFFFF_t_70

错误提示需要变量是DoubleTensor类型,但得到的却是FloatTensor类型。于是我把报错的那一行涉及到的两个变量都转为了DoubleTensor类型,果然就好了,见下图:

watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM5OTM4NjY2_size_16_color_FFFFFF_t_70 1

发表评论

表情:
评论列表 (有 0 条评论,349人围观)

还没有评论,来说两句吧...

相关阅读