有大佬能帮忙训练画风吗？或者有什么办法能解决这个问题【novelai吧】

novelai吧关注：362,375贴子：1,813,079

训练喜欢画师画风一直这个报错不知道咋解决。摆烂了

送TA礼物

IP属地:湖南

来自Android客户端1楼2022-10-12 17:01回复

你这个是输入的特征张量存储位置不对，你检查一下是不是有一个输入张量忘记加.to("cuda")了

IP属地:北京

来自Android客户端2楼2022-10-12 18:02

不感兴趣

开通SVIP免广告

初始化的tensor都是存储在cpu的，需要转存到显卡里面才能一起加载

IP属地:北京

来自Android客户端3楼2022-10-12 18:03

收起回复

同问，lz解决了吗

IP属地:上海

5楼2022-10-24 21:57

下载贴吧APP
看高清直播、视频！

贴吧热议榜

发表回复

内容:

使用签名档查看全部

发表

保存至快速回贴