使用Pytorch搭建神经网络

最新推荐文章于 2023-06-13 07:42:47 发布

Huntersts

最新推荐文章于 2023-06-13 07:42:47 发布

阅读量402

点赞数

本文链接：https://blog.csdn.net/qq_35260390/article/details/105137683

版权

本文介绍了如何使用Pytorch构建神经网络的简单框架，包括获取适合神经网络的数据batch，以及建立网络结构和优化器进行训练。核心是利用torch.utils.data.DataLoader处理数据集。

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

Pytorch搭建神经网络的一个简单框架

1.获取神经网络能用的数据（batch）
2.构建网络（具体看自己实现的网络，先不写）
3.建立优化器和训练

torch使用torch.utils.data.DataLoader读取数据，将数据集转换成最终能用的batch，一般搭建网络都需要这个。

import torch.utils.data as data
# dataset 假设dataset是原始数据集
# 先要定义一个class继承torch.utils.data.Dataset
class Mydataset(data.Dataset):
	def __init__(self

关注