2021 年 11月 27 日随笔档案 - 平凡之上

2021年11月27日

摘要： #P30-P33完结利用GPU来训练方法1：思路：找到网络模型、数据（输入，标注）、损失函数。然后调用.cuda() 然后返回方法2： Device=torch.device("CPU") torch.device("cuda:0") #指定显卡 .to(device) 阅读全文

posted @ 2021-11-27 19:09 平凡之上阅读(23) 评论(0) 推荐(0) 编辑

11/26日27日学习2

摘要： #P27-P29 完整的模型训练套路思路步骤：准备数据集→DataLoader加载数据集→创建网络模型→创建损失函数→设置优化器→设置训练的一些参数 →开始训练步骤：↓for data in train_dataloader ↓loss计算损失值 ↓优化器优化模型 → ↓with torch.n 阅读全文

posted @ 2021-11-27 16:21 平凡之上阅读(17) 评论(0) 推荐(0) 编辑