摘要: 整理一下之前的项目用到的一些函数和类,防止过几天给忘了然后看不懂自己的代码 几个比较固定的用法 device=torch.device("CUDA" if torch.cuda.is_available()==True else "cpu")#检查cuda Module.to(device) ten 阅读全文
posted @ 2026-01-29 11:32 2549866 阅读(0) 评论(0) 推荐(0)