pytorch训练常用语句
文章平均质量分 71
读博好难啊
这个作者很懒,什么都没留下…
展开
-
PyTorch常用代码段合集
提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档文章目录前言一、基本配置1.导入包和版本查询2.可复现性3.显卡设置二、张量(Tensor)处理1.张量的数据类型2.张量的基本信息3.张量的命名(神奇)4.数据类型转换5.torch.Tensor与np.ndarray转换6.Torch.tensor与PIL.Image转换7.np.ndarray与PIL.Image的转换8.从只包含一个元素的张量中提取值9.张量形变10.打乱顺序11.水平翻转12.复制张量13.张量拼接14.将整数标.原创 2022-04-28 12:40:23 · 269 阅读 · 0 评论 -
PyTorch中调用用GPU的方法
设定GPU的模块device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") 注意:仔细查看你的设备GPU的编号,我的设备禁用了集成显卡,仅留一个显卡使用。将数据发送至GPU上这部分很简单,将model、input、optimizer等数据后加*.to(device)*,例如:model.to(device)查看GPU使用情况使用任务管理器查看:打开任务管理器,在性能标签页查看使用命令行窗口原创 2021-06-16 16:14:33 · 5346 阅读 · 0 评论