Pytorch搭建入门级网络

最新推荐文章于 2024-06-23 18:54:35 发布

YuTesla

最新推荐文章于 2024-06-23 18:54:35 发布

阅读量479

点赞数

分类专栏：入门教程文章标签： python 神经网络深度学习

本文链接：https://blog.csdn.net/TelsaGao/article/details/106290518

版权

本文面向深度学习初学者，通过实例详细讲解如何在Pytorch框架下搭建网络，包括完全使用Pytorch工具和手动实现随机梯度下降（SGD）。内容涵盖网络结构、超参数设置、训练过程以及手动SGD的实现，旨在帮助新手巩固理论知识并提供实践指导。

摘要由CSDN通过智能技术生成

作为一个只学习了深度学习的理论基础，实操经验为0的小白，想通过输出的方式巩固一下搭网络的技能，并给同样是新手的同学一个比较基础的入门实例，这一篇是pytorch框架下搭网络的示例，包括使用pytorch自己的一套工具搭建，和只使用pytorch的autograd机制来搭建网络。

在阅读实例之前，推荐看一下pytorch的autograd的官方文档，或者中文翻译，有助于理解。

注：以下代码有部分参考莫烦的pytorch教程中的分类网络，有兴趣的同学也可以看看他的教程

话不多说，开始上代码。

一、完全使用pytorch框架搭建网络

import numpy as np
import matplotlib.pyplot as plt

//生成200个数据点
n_data = torch.ones(100, 2)
x0 = torch.normal(2*n_data, 1)      # class0 x data (tensor), shape=(100, 2)
y0 = torch.zeros(100)               # class0 y data (tensor), shape=(100, 1)
x1 = torch.normal(-2*n_data, 1)     # class1 x data (tensor), shape=(100, 2)
y1 = torch.ones