search

breadcake / unsupervisedDeepHomography-pytorch

Pytorch implementation of Unsupervised Deep Homography.
36 stars 5 forks source link

gpu上训练 #5

Open fanzhen12 opened 2 years ago

fanzhen12 commented 2 years ago

我在跑这个模型的时候，无法把它放在gpu上跑，请问，您使用gpu跑这个深度学习项目时，直接使用的这份代码，没有再添加其他的语句吗。我直接使用这份代码时，代码还是在cpu上运行。（我的cuda是可用的，is_available返回true ）