摘要: 在上一篇博客中,我们快速搭建,训练了一个小型的网络,但是存在一下问题。 仅仅是使用了 CPU,并没有使用 GPU 进行训练; 学习率太高,导致最后数值提不上去; 针对这2个问题,我们进行统一的解决。 并最后写一个 detect 模块,将我们写出的网络进行应用。 pytorch 使用 GPU 进行训练 阅读全文
posted @ 2021-11-13 21:30 lucky_light 阅读(894) 评论(0) 推荐(0) 编辑
摘要: 本篇博客中,我将快速搭建一个小型的网络,并对其进行训练、优化器调参,最后查看模型训练效果。 我将本次搭建网络分为一下几个部分 下载、读取数据 搭建网络 准备日志、损失函数和优化器 进行网络的训练与测试, 模型文件的保存 关闭日志并查看训练效果 下载读取数据 本篇博客所写代码使用 python,并且大 阅读全文
posted @ 2021-11-13 14:56 lucky_light 阅读(508) 评论(0) 推荐(0) 编辑
摘要: PyTorch是一个开源的Python机器学习库,基于Torch,用于自然语言处理等应用程序。他提供了大量的模型供我们所使用,如下图所示: 下面,我们选择其中一个网络进行使用,介绍如何使用、并修改 pytorch 本身为我们提供的现有网络。最后介绍一下模型的保存和修改。 pytorch 现有网络的使 阅读全文
posted @ 2021-11-13 10:11 lucky_light 阅读(660) 评论(0) 推荐(0) 编辑