why you use cpu --->model(inp).cpu() ...............................

[cqray1990]

[Donaghys]

No description provided.

我也想知道，为什么要使用cpu，如果改成mode(inp)的话非常影响网络最终的训练结果~我这边，前期loss能差两个数量级

[ingale726]

我改成了.cuda(),不知道效果怎么样

[tankgit]

作者高估了用cpu进行infer的效率。且不说效率，光训练直接拉满近百张cpu，跪了