训练问题

[teorzhang]

请问为什么我训练自己的数据集时会卡在这一步很久，没有开始训练。sample_batch: 4 torch.Size([256, 600]) torch.Size([256, 600]) torch.Size([256, 600]) torch.int64 torch.Size([256]) torch.int64。到这就不输出了

[teorzhang]

卡在这一步model = model.to(device)，去掉会报runtimeerror: module must have its parameters and buffers on device cuda:0 (device_ids[0]) but found one of them on device: cpu 。不去掉会一直卡着。。。

[zhci]

卡在这一步模型 = model.to（device），去掉会报运行时错误： 模块必须在设备上有其参数和缓冲区 cuda：0 （device_ids[0]） 但在设备上找到其中一个：CPU。不去掉会一直卡着。。。

请问你解决了么

[teorzhang]

卡在这一步模型 = model.to（device），去掉会报运行时错误： 模块必须在设备上有其参数和缓冲区 cuda：0 （device_ids[0]） 但在设备上找到其中一个：CPU。不去掉会一直卡着。。。

请问你解决了么

没，不用这个仓库代码了

[yihp]

@teorzhang 你后面用的哪个仓库兄弟